import imgaug.augmenters as iaa
import cv2
import numpy as np
# 定义数据增强方法
seq = iaa.Sequential([
# iaa.Flipud(0.5), # 纵向翻转,概率为50%
# iaa.Fliplr(0.5), # 横向翻转,概率为50%
# iaa.Affine(rotate=(-15, 15)), # 旋转,范围在-15到15度之间
iaa.GammaContrast(gamma=(0.8, 1.2)), # 伽马校正,范围在0.5到1.5之间
iaa.AdditiveGaussianNoise(scale=0.05 * 255), # 添加高斯噪声
iaa.Crop(px=(0, 16)), # 随机裁剪,每侧裁剪0到16像素
iaa.GaussianBlur(sigma=(0, 3.0)), # 高斯模糊,sigma范围在0到3.0之间
iaa.ElasticTransformation(alpha=(0, 30), sigma=5), # 弹性扭曲,alpha范围在0到30,sigma为5
iaa.Affine(translate_percent={"x": (-0.2, 0.2), "y": (-0.2, 0.2)}) # 平移,x和y方向上平移-10%到10%
])
# 定义图像特有的增强方法
s