Augmentor使用学习笔记

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_bottom(probability=0.5)

p.sample(10000) #根据你的设定，生成10000个数据增强样本，保存在output目录下。

#不想把数据保存在磁盘中，你可以使用Keras创建一个generator
g = p.keras_generator(batch_size=128)
ima