如何将具有透明度的灰度 png 转换为热图？

Question

Python 3 不支持heatmap包，对于带有 alpha 通道的 PNG 图像， cv2不支持以下内容：

cv2.applyColorMap(img, cv2.COLORMAP_JET)

我想将灰度 PNG 图像转换为热图； 换句话说，将较暗的像素着色为蓝色，将较亮的像素着色为红色。

每个像素的透明度应保持不受影响。

Answer 1

尝试使用 matplotlib.pyplot.get_cmap。

colormap = plt.get_cmap('plasma')
heatmap = (colormap(image) * 2**16).astype(np.uint16)[:,:,:3]
heatmap = cv2.cvtColor(heatmap, cv2.COLOR_RGB2BGR)

您可以根据所需的输出选择颜色图。