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

Question

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

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)

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