使用 Augmentor 批量化进行数据增强

一、在批量化生成数据集前需要做的事情：

1.将.jpg格式转化为.png格式。

这里可以参考上一篇博客，批量化的进行图片格式的转换.JPG图片批量化转换为.PNG图片_Wang0613_的博客-CSDN博客

2. pip install Augmentor

二、使用以下代码进行数据数据增强

# 导入数据增强工具
import Augmentor

# 确定原始图像存储路径以及掩码文件存储路径
p = Augmentor.Pipeline("D:/UserData/Desktop/1/新增jpg")
p.ground_truth("D:/UserData/Desktop/1/png")

# 图像旋转： 按照概率0.8执行，最大左旋角度10，最大右旋角度10
p.rotate(probability=0.8, max_left_rotation=15, max_right_rotation=15)

# 图像左右互换： 按照概率0.5执行
p.flip_left_right(probability=0.5)

# 图像上下互换，按照概率0.5执行
p.flip_top_bottom(probability=0.5)

# 图像放大缩小： 按照概率0.8执行，面积为原始图0.85倍
p.zoom_random(probability=0.3, percentage_area=0.85)

# 随机擦除与遮挡
p.random_erasing(probability=1, rectangle_area=0.5)

# 最终扩充的数据样本数
p.sample(480)

p = Augmentor.Pipeline("D:/UserData/Desktop/1/新增jpg")
p.ground_truth("D:/UserData/Desktop/1/png")

需要把以上两个路径进行修改，即第一为原图文件，第二为对应的标签文件。 

此代码会在原图文件中生成一个output 文件夹，其中包括生成的原图和对应的掩码图。

参考博客：

语义分割 图像增强（数据扩充）_图像增强能做为数据扩充的方法吗_钱彬 (Qian Bin)的博客-CSDN博客