如何获取多输出模型的类激活映射?

2021-12-01 12:15:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图可视化CNN模型的每个输出的Gradcam激活图,带有多个输出。问题是如何实现“损失函数”?我指的是这个代码

更清楚地说,问题是如何可视化output1、output2、output3的渐变图

enter image description here

范例 定义损失函数: 必须定义返回目标分数的损失函数。在这里,它返回相应的分数 金鱼,熊,突击步枪

# The `output` variable refer to the output of the model,
# so, in this case, `output` shape is `(3, 1000)` i.e., (samples, classes).

def loss(output):
    return (output[0][1], output[1][294], output[2][413])
# 1 is the imagenet index corresponding to Goldfish, 294 to Bear and 413 to Assault Rifle.