语义分割图像和标签一起扩充

        网上很多数据扩充都是只扩充了图像,图像和对应的标签一起扩充的代码很少,而且很多代码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 文件夹放的就是扩充后的图像数据和对应的标

注意:图像和对应的标签一定要名称完全一致(包括后缀)才能使用这段代码完成数据扩充。不然就只会扩充图像,而不会扩充标签

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值