语义分割数据扩充(图像和标签同步扩充)

7 篇文章 2 订阅
2 篇文章 0 订阅

发现一个python包Augmentor,专门用于数据扩充,链接:https://augmentor.readthedocs.io/en/master/userguide/install.html,就是开发手册,里面包含了安装,包内函数的介绍、扩展性等,这个模块主要包括了:随机旋转、随机裁剪、镜像、随机变形、随机亮度、随机颜色、随机对比度、随机擦除等等。应该能够满足要求了,不满足还可以自己加函数用它的壳子套上自己用。

1.这个包的使用比较简单,参数的具体解释可以在链接手册上查看,网上已经有很多介绍的了,下面copy一些代码来稍作说明
#导入数据增强工具
import Augmentor

p = Augmentor.Pipeline("test1")  #test1为图像文件路径,最后的输出会在这个文件下新建一个output文件夹并存入
p.ground_truth("test2")   #test2为标签文件路径
#图像旋转: 按照概率0.8执行,最大左旋角度10,最大右旋角度10
p.rotate(probability=0.8, max_left_rotation=10, max_right_rotation=10)
#图像左右互换: 按照概率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(20)

注意:输出的结果都是一一对应的,只不过名字有点混乱,还需要自己写代码重新整理一下

2.下面就简单说一下扩展性。这个模块允许我们自己写一些扩充函数去补充已有函数的不足。

这里的FoldImage就是你自己写的扩充函数类,继承了Operation,perform_operation函数就你的具体实现,具体使用和其他自带函数一样的方式

3.手册中还有提到这个数据扩充和Keras一起使用喔

  • 3
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

如雾如电

随缘

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值