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

# 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_img

 def 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_img

 def 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.2
    image_contrasted = enh_con.enhance(contrast)
    return image_contrasted

def brightnessEnhancement(root_path,img_name):#亮度增强
    image = Image.open(os.path.join(root_path, img_name))
    enh_bri = ImageEnhance.Brightness(image)
    brightness = 1.3
    image_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.5
     image_colored = enh_col.enhance(color)
     return image_colored

from PIL import Image
from PIL import ImageEnhance
import os
import cv2
import numpy as np
imageDir="D:/..."     #要改变的图片的路径文件夹
saveDir="D:/..."   #要保存的图片的路径文件夹

for name in os.listdir(imageDir):
    saveImage=contrastEnhancement(imageDir,name)
    saveName = name
    saveImage.save(os.path.join(saveDir,saveName))

     saveImage=brightnessEnhancement(imageDir,name)
     saveName = name
     saveImage.save(os.path.join(saveDir,saveName))

     
     saveImage=flip(imageDir,name)
     saveName = name
     saveImage.save(os.path.join(saveDir,saveName))
    
     
     saveImage=rotation(imageDir,name)
     saveName = name
     saveImage.save(os.path.join(saveDir,saveName))