图像分类中的图像增强，批量对某一张图像增强，生成多张图像

**注意修改这两个路径，**指定原始图像文件夹和输出文件夹

input_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\origin_bmp\lunxing\09523C01"
output_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\origin_bmp\lunxing\09523C01_aug"
*另外注意图像必须是RGB彩色三通道的，灰度图会报错

设置增强后的图像数量，这里可以指定要生成的图像的数量

num_samples = 100
#############################################以下是代码

import Augmentor

# 指定原始图像文件夹和输出文件夹
input_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\origin_bmp\lunxing\09523C01"
output_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\origin_bmp\lunxing\09523C01_aug"

# 创建 Augmentor 队列
p = Augmentor.Pipeline(input_folder, output_folder)

# 在队列中添加数据增强操作
p.rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
p.flip_left_right(probability=0.5)
p.zoom_random(probability=0.5, percentage_area=0.8)
p.random_contrast(probability=0.5, min_factor=0.5, max_factor=1.5)

# 随机旋转
p.rotate_random_90(probability=0.5)

# 随机缩放
p.zoom_random(probability=0.5, percentage_area=0.8)
# p.zoom_random(probability=0.5)

# 随机平移
p.random_distortion(probability=0.5, grid_width=4, grid_height=4, magnitude=8)

# 随机添加噪声
p.random_contrast(probability=0.5, min_factor=0.5, max_factor=1.5)
p.random_brightness(probability=0.5, min_factor=0.5, max_factor=1.5)
p.random_color(probability=0.5, min_factor=0.5, max_factor=1.5)

# 随机模糊
# p.gaussian_blur(probability=0.5, min_sigma=0.1, max_sigma=1)

# 随机变换亮度
p.random_brightness(probability=0.5, min_factor=0.7, max_factor=1.3)

# 随机变换对比度
p.random_contrast(probability=0.5, min_factor=0.7, max_factor=1.3)

# 随机擦除一部分区域
p.random_erasing(probability=0.5, rectangle_area=0.2)

# 添加随机高斯噪声
# p.random_noise(probability=0.5, var=0.1)

# 随机水平翻转
p.flip_left_right(probability=0.5)

# 随机垂直翻转
p.flip_top_bottom(probability=0.5)


# 设置增强后的图像数量
num_samples = 100

# 执行数据增强
p.sample(num_samples)