基于python绘制热力图

        这里介绍两种热力图绘制方法，可以用于区域的概率显示。

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的热力图

热力图半透明掩膜