好用的数据增强工具Augmentor(原图和mask扩充)

  • Post author:
  • Post category:其他




1.原图和mask扩充增强

# 导入数据增强工具
import Augmentor

# 确定原始图像存储路径以及掩码文件存储路径
p = Augmentor.Pipeline("image200") # 原图和mask的标签要一一对应!!扩充后的图都在原图里面新生成output的文件夹!
p.ground_truth("mask200") #标签图mask

# 图像旋转: 按照概率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)

# 最终扩充的数据样本数(200到2000张,扩充了十倍)
p.sample(2000)                    

在这里插入图片描述

在这里插入图片描述



版权声明:本文为weixin_44245653原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。