生成一张图片的热力图

xiao_ju_zi_

于 2024-12-30 16:33:18 发布

阅读量147

点赞数 4

文章标签： python

本文链接：https://blog.csdn.net/xiao_ju_zi_/article/details/144829328

版权

import cv2
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm

# 读取图片
image_path = r"C:\Users\24909\Desktop\shujuji\train\2\20221002_0801_a7e8e904-41fd-4720-91b5-d1353c89caf6_flip_translated.jpg"
img = cv2.imread(image_path)

# 转换为灰度图
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 生成热力图
heatmap = cv2.applyColorMap(gray_img, cv2.COLORMAP_JET)

# 显示热力图
plt.imshow(heatmap)
plt.axis('off')  # 不显示坐标轴
plt.show()

# 保存热力图
cv2.imwrite("heatmap_output.jpg", heatmap)