Augmentor是个增强图像训练数据的库,减少了使用图像库自己编写代码的繁杂工序,能够批量完成图像的旋转,放大,缩小,添加噪音以扩充数据量。接下来结合官方文档介绍下这个库和使用心得。
首先github:https://github.com/mdbloice/Augmentor
DOCs:https://augmentor.readthedocs.io/en/master/userguide/mainfeatures.html#perspective-skewing
安装:
pip install Augmentor
升级: pip install Augmentor --upgrade
导入:(导入需要增强数据的目录,这里以该路径下的100张图像初始化)
import Augmentor p = Augmentor.Pipeline("/path/to/images")然后你就可以使用这个pipeline对象p进行图像变换了:
在执行完所需的处理操作后通过:
p.sample(10000)
来进行样本的输出(在路径目录上新生成的output文件夹中)
首先是基本的旋转:
p.rotate(probability=0.7, max_left=10, max_right=10)
输入图像 | 输出图像例 |