用opencv查看图像的attention,形象观看效果,可以利用cv2.applyColorMap来生成图像。
卷积中系数一般是1,直接相加。
python代码如下:
import cv2
import numpy as np
img = cv2.imread('img.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 # img为前景,0.6为背景
x = x.astype(np.uint8)
cv2.imwrite('hhh.jpg',x)