【matplotlib 叠加图片显示热力图 attention map】

有的时候需要看attention map集中在图片当中的哪些区域，或者需要看重构误差在哪些区域比较大，所以需要将“热力图”和图片叠加显示，先看效果图：

                                  

代码如下：

import matplotlib.pyplot as plt

img_path = 'IMAGE_PATH'
mask_path = 'MASK_PATH'

img = plt.imread(img_path)
mask = plt.imread(mask_path)

#叠加显示img, mask
plt.imshow(img)
plt.imshow(mask, alpha=0.4, cmap='rainbow')  #alpha设置透明度, cmap可以选择颜色

plt.imshow()