数据增强-翻转 旋转 随机颜色 对比度增强 亮度增强 颜色增强

# imageDir 为原数据集的存放位置# saveDir 为数据增强后数据的存放位置####def flip(root_path,img_name): #翻转图像img = Image.open(os.path.join(root_path, img_name))filp_img = img.transpose(Image.FLIP_LEFT_RIGHT)filp_img.save(os.path.join(root_path,img_name.split('.')[0] + '_flip.jpg'))return filp_imgdef rotation(root_path, img_name):img = Image.open(os.path.join(root_path, img_name))rotation_img = img.rotate(20) #旋转角度rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.jpg'))return rotation_imgdef randomColor(root_path, img_name): #随机颜色"""对图像进行颜色抖动:param image: PIL的图像image:return: 有颜色色差的图像image"""image = Image.open(os.path.join(root_path, img_name))random_factor = np.random.randint(0, 31) / 10. # 随机因子color_image = ImageEnhance.Color(image).enhance(random_factor) # 调整图像的饱和度random_factor = np.random.randint(10, 21) / 10. # 随机因子brightness_image = ImageEnhance.Brightness(color_image).enhance(random_factor) # 调整图像的亮度random_factor = np.random.randint(10, 21) / 10. # 随机因子contrast_image = ImageEnhance.Contrast(brightness_image).enhance(random_factor) # 调整图像对比度random_factor = np.random.randint(0, 31) / 10. # 随机因子return ImageEnhance.Sharpness(contrast_image).enhance(random_factor) # 调整图像锐度def contrastEnhancement(root_path, img_name): # 对比度增强image = Image.open(os.path.join(root_path, img_name))enh_con = ImageEnhance.Contrast(image)contrast = 2.2image_contrasted = enh_con.enhance(contrast)return image_contrasteddef brightnessEnhancement(root_path,img_name):#亮度增强image = Image.open(os.path.join(root_path, img_name))enh_bri = ImageEnhance.Brightness(image)brightness = 1.3image_brightened = enh_bri.enhance(brightness)return image_brightened#对比度 contrast的最佳取值范围在[0 ~ 4],亮度 brightness的最佳取值范围在[0~ 2]之间def colorEnhancement(root_path,img_name):#颜色增强image = Image.open(os.path.join(root_path, img_name))enh_col = ImageEnhance.Color(image)color = 1.5image_colored = enh_col.enhance(color)return image_coloredfrom PIL import Imagefrom PIL import ImageEnhanceimport osimport cv2import numpy as npimageDir="D:/..."#要改变的图片的路径文件夹saveDir="D:/..." #要保存的图片的路径文件夹for name in os.listdir(imageDir):saveImage=contrastEnhancement(imageDir,name)saveName = namesaveImage.save(os.path.join(saveDir,saveName))saveImage=brightnessEnhancement(imageDir,name)saveName = namesaveImage.save(os.path.join(saveDir,saveName))saveImage=flip(imageDir,name)saveName = namesaveImage.save(os.path.join(saveDir,saveName))saveImage=rotation(imageDir,name)saveName = namesaveImage.save(os.path.join(saveDir,saveName))