网上很多数据扩充都是只扩充了图像,图像和对应的标签一起扩充的代码很少,而且很多代码copy过来也不能直接用。
#导入数据增强工具
import Augmentor
#确定原始图像存储路径以及掩码文件存储路径
p = Augmentor.Pipeline("D:\folder1") #这里填写图像文件路径
p.ground_truth("D:\folder2") #这里填写图像对应的标签文件路径
#图像旋转: 按照概率0.8执行,最大左旋角度25,最大右旋角度25
p.rotate(probability=0.8, max_left_rotation=25, max_right_rotation=25)
#图像左右互换: 按照概率0.5执行
p.flip_left_right(probability=0.5)
#图像放大缩小: 按照概率0.8执行,面积为原始图0.85倍
p.zoom_random(probability=0.3, percentage_area=0.85)
#最终扩充的数据样本数
p.sample(50) #想要扩充到多少张,这里50表示扩充后一共有50张(不包括原来待扩充的图像)
原先crack / image文件夹下只有18张图像
运行完成过后,会在原先图像路径crack / image路径里面自动新建一个output文件夹
output 文件夹放的就是扩充后的图像数据和对应的标
注意:图像和对应的标签一定要名称完全一致(包括后缀)才能使用这段代码完成数据扩充。不然就只会扩充图像,而不会扩充标签
标签:25,标签,路径,语义,50,图像,扩充 From: https://blog.csdn.net/qq_54434270/article/details/140567863