数据增强库Augmentor使用，以及批量复制到指定文件，删除原文件

安装：pip install Augmentor
设置路径：p=Augmentor.Pipeline(yourpath)
设置所需的变化：
probability为图片的比例
p.rotate90(probability=1.0) 旋转90°
p.rotate(probability=0.5,max_left_rotation=25,max_right_rotation=25)
p.skew_tilt(probability=0.7,magnitude=1)
其余参考网址1：https://blog.csdn.net/qq_34374211/article/details/83280215
参考网址2：https://blog.csdn.net/timcanby/article/details/79247603

设置需要的数量：
p.sample(200)

生成的路径会在指定文件的output文件夹下，这就需要批量复制图片，指定文件名以及删除原来的文件。

import os
import Augmentor
import shutil

def augment():
    path = r'E:/sign_system/crop'
    dirs = os.listdir(path)
    for label in dirs:
        if not label :
            break
        pic_path = path + '/'+ str(label)
        print(pic_path)
        p = Augmentor.Pipeline(pic_path)
        p.flip_left_right(probability=1)#水平镜像
        p.rotate(probability=1,max_left_rotation=10,max_right_rotation=10)#随机旋转10°
        #变化还可以继续添加
        p.sample(10)
        
def copy():
    path = r'E:/sign_system/crop'
    dirs = os.listdir(path)
    for label in dirs:
        if not label :
            break
        #原文件里有5张图片（0~4），所以图片名从5开始     
        num = 5
        pic_path = path + '/' + str(label) + '/output'
        new_path = path + '/' + str(label)
        pic_dirs = os.listdir(pic_path)
        for pic in pic_dirs: 
        	#新的图片名
            new_img = '%s/%d.jpg'%(new_path, num)
            shutil.copy(pic_path+'/'+pic,new_img)
            num = num +1
        shutil.rmtree(pic_path)
        #注意使用os.remove会报错，（winerror 5）
if __name__ == "__main__":
    augment()
    copy()

这样就大功告成啦！！！！