opencv_python实现批量图片颜色变换,可用于数据增强

变换原理:HSV

算法流程:cv2依次读取文件下图片rgb值,再转为hsv,通过改变hsv中的h(即色调)值改变图片的颜色,生成多张颜色变换后的图片(可用于数据增强)。

废话不多说,上代码(用的时候记得将两个文件夹路径改为自己的输入和输出路径,文件夹下的图片按1,2,3,…的顺序命名):

import cv2

num = 0 #读取的图片序号 
num_max = 7 #图片总数量
hue_change = 5 #色调改变值 步长
count = 0 #记录每张图片生成的数量

while 1:
    img_name = 'C:/Users/hq/Desktop/person_img/img/%s.jpg' % str(num+1)
    img=cv2.imread(img_name, cv2.IMREAD_COLOR)    # 打开文件
     
    # 通过cv2.cvtColor把图像从BGR转换到HSV
    img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
     

    turn_green_hsv = img_hsv.copy()
    # 色调hue变化范围5——50
    for i in range(1,11):
        turn_green_hsv[:, :, 0] = (turn_green_hsv[:, :, 0] + hue_change*i) % 180
        turn_green_img = cv2.cvtColor(turn_green_hsv, cv2.COLOR_HSV2BGR)

        cv2.imwrite('C:/Users/hq/Desktop/person_img/color_change_img/%s_%s.jpg'%(str(num+1), str(count+1)), turn_green_img)
        print("successfully save %s_%s pic" %(str(num+1), str(count+1)))
        count += 1

    if num == num_max - 1:
        exit(0)
    
    count = 0
    num += 1

运行结果:


这是笔者的输入文件夹下的图片:
opencv_python实现批量图片颜色变换,可用于数据增强_第1张图片
这是输出文件夹下的生成结果:
opencv_python实现批量图片颜色变换,可用于数据增强_第2张图片
注:笔者的预设色调变换值是5~50,各位读者可以根据需求,自己设置h的步长以及范围。

你可能感兴趣的:(数据增强)