python cv2 热度图

生成热度图可以使用Python中的OpenCV库（cv2）和NumPy库。下面是一个简单的例子，用于从图像中提取热度图。

假设你已经有一个灰度图像作为输入。首先，需要将图像转换为伪彩色映射（color map）（也称为“热力图”）。这可以通过cv2.applyColorMap()函数来实现。然后，将伪彩色映射与原始灰度图像加权相加，以创建热度图像。

import cv2
import numpy as np

# 读取灰度图像
gray = cv2.imread('input.jpg', cv2.IMREAD_GRAYSCALE)

# 应用热力图
heatmap = cv2.applyColorMap(gray, cv2.COLORMAP_JET)

# 将热力图与原始图像混合
alpha = 0.5
output = cv2.addWeighted(heatmap, alpha, gray, 1 - alpha, 0)

# 保存输出图像
cv2.imwrite('output.jpg', output)

在上面的代码中，使用的是cv2.COLORMAP_JET颜色映射，你可以根据需要选择其他颜色映射。最终输出的图像保存为output.jpg。