Augmentor介绍:Augmentor是一个Python包,旨在帮助机器学习任务的图像数据人工生成和数据增强。它主要是一种数据增强工具,但也将包含基本的图像预处理功能。
github地址链接:https://github.com/mdbloice/Augmentor
安装:pip install Augmentor
更新:pip install Augmentor --upgrade
Augmentor:图像数据扩增工具,独立于平台和框架。
自动化的进行图像增强。
可选择的操作:
Augmentor创建了一个图像扩增的pipeline,可以在上面定义一系列操作。每个操作对应一个概率,用来决定是否对图像进行处理。
使用:
初始化一个Pipeline object,使其指向一个目录。
import Augmentor
#创建pipeline,并指定参数
p = Augmentor.Pipeline("/home/tsq/MobileNet/Picture")
p.rotate(probability=0.7,max_left=10,max_right=10) #旋转操作
p.zoom(probability=0.5,min_scale=1.1,max_scale=1.5) #缩放操作
p.random_distortion(probability=1, grid_width=4, grid_height=4, magnitude=8)
p.flip_left_right(probability=0.5)
p.flip_top_bot