python批量将jpg图片转换成热力图保存并显示

code•2022-4-9•python•阅读63

import cv2
import matplotlib.pyplot as plt
import os

path ="jpg文件地址/"
dir = os.listdir(path)
for a in dir:
    stem, suffix = os.path.splitext(a)
    image=cv2.imread(str(path)+str(a),2)
    image1=cv2.applyColorMap(image, cv2.COLORMAP_JET)

    # image1 = cv2.cvtColor(image,cv2.COLOR_GRAY2BGR)
    cv2.imwrite("保存地址/"+stem+'.jpg',image1)
    plt.imshow(image1)
    plt.show()