变换原理: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
运行结果:
这是笔者的输入文件夹下的图片:
这是输出文件夹下的生成结果:
注:笔者的预设色调变换值是5~50,各位读者可以根据需求,自己设置h的步长以及范围。