applyColorMap を使用すると、単一チャンネルの画像に対して擬似カラー処理とヒートマップを適用することができます。OpenCV では 12 種類のカラーマップ定数が定義されており、必要なものを選択することができます。cv2.applyColorMap (heatmap_g, cv2.COLORMAP_JET) 画像は addWeighted を使用して重ね合わせ処理することができます。cv2.addWeighted (heatmap_img, alpha, merge_img, 1-alpha, 0, merge_img) # ヒートマップを元の画像にオーバーレイする
def heatmap_overlay(image,heatmap):
# ヒートマップをグレースケールに変換
heatmap_g = heatmap.astype(np.uint8)
# ヒートマップを擬似カラーに変換
heatmap_color = cv2.applyColorMap(heatmap_g, cv2.COLORMAP_JET)
# ヒートマップをオーバーレイ
merge_img = image.copy()
heatmap_img = heatmap_color.copy()
overlay = image.copy()
alpha = 0.25 # オーバーレイ画像の透明度を設定
#cv2.rectangle (overlay, (0, 0), (merge_img.shape [1], merge_img.shape [0]), (0, 0, 0), -1) # ヒートマップの基本色を青に設定
cv2.addWeighted (overlay, alpha, merge_img, 1-alpha, 0, merge_img) # 背景のヒートマップを元の画像にオーバーレイ
cv2.addWeighted (heatmap_img, alpha, merge_img, 1-alpha, 0, merge_img) # ヒートマップを元の画像にオーバーレイ
return merge_img
参考： https://blog.csdn.net/u013381011/article/details/78341861