使用opencv+python 画出attention map

有时候需要对图像的attention可视化，形象观看效果，可以利用cv2.applyColorMap来生成图像。

输入：原图、与原图同样大小的attention map

python代码如下：

import cv2
import numpy as np

img = cv2.imread('demo.jpg', 1)    # 原图
att = cv2.imread('demo_att.jpg',1)    # feature map
w = cv2.applyColorMap(att, 2)    # 转化为jet 的colormap
x = img * 0.4 + w * 0.6    # 权重自己定
x = x.astype(np.uint8)
cv2.imwrite('hhh.jpg',x)

这里随便手画了一个feature map channel，以供示例