admin管理员组文章数量:1650768
显示feature的heatmap
import cv2
import numpy as np
import scipy.misc
# feature：numpy array; 从网络的forward过程当中保存下来的feature map
# row_image：numpy array; 3*224*224
# output_jpg_name: eg. "out.jpg"
def show_heatmap(feature,output_jpg_name,row_image):
data = feature
heatmap = data.sum(0)/data.shape[0]
heatmap = np.maximum(heatmap, 0)
heatmap /= np.max(heatmap)
heatmap = 1.0 - heatmap # 也可以不写，就是蓝色红色互换的作用
heatmap = cv2.resize(heatmap, (224,224)) # (224,224)指的是图像的size，需要resize到原图大小
heatmap = np.uint8(255 * heatmap)
heatmap = cv2.applyColorMap(heatmap, cv2.COLORMAP_JET)
row = row_image
row = row.transpose(1,2,0)
superimposed_img = heatmap*1.0 + row*0.5 # 1.0 和 0.5代表heatmap和row image的强度占比，可调整
scipy.misc.imsave(output_jpg_name,superimposed_img)
版权声明：本文标题：显示feature的heatmap热力图 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://www.elefans.com/xitong/1729533056a1205057.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。
发表评论