使用cv2.applyColorMap画热力图

# cam1 是[c, h, w]的矩阵ss = np.min(cam1, axis=1)bb = np.max(cam1, axis=1)for i in range(len(cam1)):for j in range(len(cam1[i])):# 不能乘以255，因为一旦x255，则热力中心部分的值就是255了，也就是白色，导致热力图中间被镂空了cam1[i][j] = (cam1[i][j

masterShaw

5348人浏览 · 2021-04-13 15:55:38

masterShaw · 2021-04-13 15:55:38 发布

# cam1 是[c, h, w]的矩阵

ss = np.min(cam1, axis=1)
bb = np.max(cam1, axis=1)
for i in range(len(cam1)):
    for j in range(len(cam1[i])):
        # 不能乘以255，因为一旦x255，则热力中心部分的值就是255了，也就是白色，导致热力图中间被镂空了
        cam1[i][j] = (cam1[i][j]-ss[i]) * 200.0 / bb[i]  

cam1 = cam1.reshape((c, h, w))
cam_img = np.uint8(cam1)

img = []
for i in range(size_upsample[0]):
    im = []
    for j in range(size_upsample[1]):
        im.append([255, 255, 255])
    img.append(im)
img = np.asarray(img)

for i in range(len(cam_img)):
    heatmap = cv2.applyColorMap(cv2.resize(cam_img[i], (size_upsample[1], size_upsample[0])), cv2.COLORMAP_JET)  # dsize = [w, h]
    heatmap = heatmap * 0.3 + img * 0.3
    cv2.imwrite("context_img/"+str(i)+".jpg", heatmap)

* 200的结果