如何將具有透明度的灰度 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)

您可以根據所需的輸出選擇顏色圖。