批量转换不同种类的光伏灰度图像为hot类型的假彩色图像

最新推荐文章于 2024-06-27 22:50:16 发布

原创最新推荐文章于 2024-06-27 22:50:16 发布 · 128 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #人工智能

这段代码实现了批量将灰度图片转换为假彩色图片，使用了OpenCV库中的applyColorMap函数，转换后的图片保存在指定目录下。支持自定义假彩色映射类型，并对多张图片进行处理。

import cv2
import glob
import os

# 多张图片的假彩色转换
up_path = r"C:\Users\sun\Desktop\data devide"
up_newpath = r"C:\Users\sun\Desktop\data devide_color"
list = os.listdir(up_path)
for lists in list:
    path = os.path.join(up_path, lists)
    newpath = os.path.join(up_newpath, lists)

    Filelist = os.listdir(path)
    a = 1
    for jpg_path in Filelist:
        Olddir = os.path.join(path, jpg_path)
        # 注意这里在im_gray输入路径时不在需要“”，因为路径本身已经有“”
        im_gray = cv2.imread(Olddir, cv2.IMREAD_GRAYSCALE)
        # filetype = os.path.splitext(Olddir)[1]
        picture_name = jpg_path

        # 后面的假彩颜色类型可以调整，这里用的是HOT
        im_color = cv2.applyColorMap(im_gray, cv2.COLORMAP_HOT)
        # 注意这里的str一定要加上
        # newdir = os.path.join(newpath, "cellcolor" + str(a) + filetype)
        newdir = os.path.join(newpath, picture_name)
        cv2.imwrite(newdir, im_color)
        a = a + 1


# 单张图片的转换
# im_gray = cv2.imread("C:/Users/sun/Desktop/new/Cell.1409.jpg", cv2.IMREAD_GRAYSCALE)
# im_color = cv2.applyColorMap(im_gray, cv2.COLORMAP_JET)
# cv2.imwrite('C:/Users/sun/Desktop/new/color/cellcolor.jpg', im_color)