opencv - 在基础图像 OpenCV Python 上叠加热图

Question

请看这个github page .我想使用 Python PIL、open cv 或 matplotlib 库以这种方式生成热图。有人可以帮我弄清楚吗？

我可以为我的网络创建一个与输入大小相同的热图，但我无法叠加它们。热图形状为 (800,800)，基本图像形状为 (800,800,3)

Answer 1

更新答案 -- 2022 年 4 月 29 日。

在反复评论之后，我决定用更好的可视化来更新这篇文章。

考虑下图:

img = cv2.imread('image_path')

我对LAB转换图像的a channel 进行二值阈值处理后得到了一个二值图像:

lab = cv2.cvtColor(img, cv2.COLOR_BGR2LAB)
a_component = lab[:,:,1]
th = cv2.threshold(a_component,140,255,cv2.THRESH_BINARY)[1]

应用高斯模糊:

blur = cv2.GaussianBlur(th,(13,13), 11)

生成的热图:

heatmap_img = cv2.applyColorMap(blur, cv2.COLORMAP_JET)

最后，将热图叠加在原始图像上:

super_imposed_img = cv2.addWeighted(heatmap_img, 0.5, img, 0.5, 0)

注意:您可以在函数 cv2.addWeighted 中改变权重参数并观察差异。