使用cv2.applyColorMap和神经网络产生的特征绘制注意力图

解决的问题

使用神经网络分类器前得到的特征绘制注意力图

示例

使用cv2.applyColorMap和神经网络产生的特征绘制注意力图_第1张图片
使用cv2.applyColorMap和神经网络产生的特征绘制注意力图_第2张图片

# 按通道压缩,feat1是分类器后的,feat2是分类器前的特征
                    with torch.no_grad():
                        feat_val, outputs_val = model(inputs)
                        feat1 = feat_val[-1]
                        feat1 = np.array(feat1.cpu

你可能感兴趣的:(个人学习_研究生,神经网络,人工智能,深度学习)