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

最新推荐文章于 2022-09-11 19:17:52 发布

yyfhq

最新推荐文章于 2022-09-11 19:17:52 发布

阅读量86

点赞数

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/yyfhq/article/details/123525821

版权

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)