Data Arguementation——数据增强（用上下翻转、左右翻转、亮度增强、对比度增强、高斯和椒盐噪声等解决数据集数量较小问题）

1.图片批量添加高斯噪声(matlab代码)

for n=0:68   %图片数量，图片命名以0.jpg、1.jpg递增
 
str1='C:\Users\cyh\Desktop\first\zunshengfomu\';   %原图片路径
str2=num2str(n);
str3='.jpg';
SC=[str1,str2,str3];
            
i=imread(SC);  
j = imnoise(i,'gaussian',0,0.006); %gauss
%imshow(i),title('Original');  
figure(2),imshow(j,'border','tight','InitialMagnification','fit'),title('Translated') ; 
set (gcf,'Position',[0,0,500,500]);
axis normal;
saveas(figure(2),['C:\Users\cyh\Desktop\gaosi\zunshengfomu\',num2str(n),'.jpg']);   %要保存的路径
end

2.图片批量添加椒盐噪声（matlab代码)

for n=0:68   %图片数量，图片命名以0.jpg、1.jpg递增
 
str1='C:\Users\cyh\Desktop\first\zunshengfomu\';   %原图片路径
str2=num2str(n);
str3='.jpg';
SC=[str1,str2,str3];
            
i=imread(SC);  
j = imnoise(i,'salt & pepper',0.008); %椒盐
%imshow(i),title('Original');  
figure(2),imshow(j,'border','tight','InitialMagnification','fit'),title('Translated') ; 
set (gcf,'Position',[0,0,500,500]);
axis normal;
saveas(figure(2),['C:\Users\cyh\Desktop\jiaoyan\zunshengfomu\',num2str(n),'.jpg']);   %要保存的图片路径
end

3.图片批量进行亮度增强、颜色增强、对比度增强（Python代码）

from PIL import Image

from PIL import ImageEnhance

import os
import PIL.Image as img
import cv2

import numpy as np

imageDir="C:/Users/cyh/Desktop/first/baidumu/" #要改变的图片的路径文件夹

saveDir="C:/Users/cyh/Desktop/1/baidumu/"   #要保存的图片的路径文件夹

filelist = os.listdir(imageDir)

def brightnessEnhancement(root_path,img_name):#亮度增强

    image = Image.open(os.path.join(root_path, img_name))

    enh_bri = ImageEnhance.Brightness(image)

    brightness = 1.5   #变暗可以较小数字，如0.5

    image_brightened = enh_bri.enhance(brightness)

    return image_brightened



def colorEnhancement(root_path,img_name):#颜色增强

    image = Image.open(os.path.join(root_path, img_name))

    enh_col = ImageEnhance.Color(image)

    color = 1.8

    image_colored = enh_col.enhance(color)

    return image_colored



def contrastEnhancement(root_path, img_name):  # 对比度增强
    image = Image.open(os.path.join(root_path, img_name))
    enh_con = ImageEnhance.Contrast(image)
    contrast = 1.5
    image_contrasted = enh_con.enhance(contrast)
    return image_contrasted


i=0

for subdir in filelist:
    sub_dir = imageDir + '/' + subdir

    im = img.open(sub_dir)

    saveImage=contrastEnhancement(imageDir,subdir)

    saveImage.save(saveDir + '/' + subdir)

4.图片批量进行上下翻转、左右翻转、旋转（Python代码）

'''
Python批量实现图像水平、垂直翻转
函数功能：扩大数据量
'''
import PIL.Image as img
import os

path_old = r"C:/Users/cyh/Desktop/first/zunshengfomu/"
path_new = r"C:/Users/cyh/Desktop/shangxia/zunshengfomu/"
filelist = os.listdir(path_old)
total_num = len(filelist)
print(total_num)

for subdir in filelist:
    sub_dir = path_old + '/' + subdir

    im = img.open(sub_dir)
    ng = im.transpose(img.ROTATE_180) #旋转 180 度角。
    # ng=im.transpose(img.FLIP_LEFT_RIGHT) #左右对换。
    # ng=im.transpose(img.FLIP_TOP_BOTTOM) #上下对换。
    # ng=im.transpose(Image.ROTATE_270) #旋转 270 度角。
   # ng = im.rotate(45)  # 逆时针旋转 45 度角。

    ng.save(path_new + '/' + subdir)

print('done')