赞
踩
这里介绍两种热力图绘制方法,可以用于区域的概率显示。
1.基于matplotlib
这里的data是一个array,读者可以自行添加。基于matplotlib的热力图有坐标,虽然方便查看,但是没办法后期叠加在原图上。
- import matplotlib.pyplot as plt
-
- cm = plt.cm.get_cmap('jet')
- plt.imshow(data, cmap=cm)
- plt.colorbar()
- plt.savefig(path+"_heatmap.png")
基于matplotlib的热力图
2.基于cv2的热力图生成
- import cv2
- #生成热力图
- heatmap = cv2.applyColorMap(np.uint8(data), cv2.COLORMAP_JET)
- #生成热力图半透明掩膜
- dst = cv2.addWeighted(img_ori,0.6,heatmap,0.4,0)
基于cv2的热力图
热力图半透明掩膜
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。