Augmentor: 强大的图像增强库

Augmentor:强大的图像增强库

Augmentor是一个用Python编写的图像增强库,专为机器学习任务而设计。它旨在成为一个独立的、平台和框架无关的库,提供更便捷、更精细的增强控制,并实现最实用的增强技术。Augmentor采用随机方法,使用构建块将操作组合成管道,从而实现灵活的图像增强。

Augmentor Logo

主要特性

独立性: Augmentor是一个独立的库,不依赖于特定的深度学习框架。
灵活的管道: 通过构建增强管道,可以轻松组合多种操作。
丰富的操作: 提供旋转、缩放、翻转、扭曲等多种图像处理操作。
概率控制: 每个操作都可以设置概率,实现随机增强。
多线程支持: 利用多线程加速图像生成过程。
Ground Truth支持: 可以同时增强原始图像和标注数据。
与主流框架集成: 提供与Keras、PyTorch等框架的集成接口。

安装

Augmentor可以通过pip轻松安装:

pip install Augmentor

快速入门

以下是一个简单的示例,展示了如何使用Augmentor创建增强管道:

import Augmentor

# 创建Pipeline对象
p = Augmentor.Pipeline("/path/to/images")

# 添加操作
p.rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
p.zoom(probability=0.5, min_factor=1.1, max_factor=1.5)

# 生成样本
p.sample(10000)

这个简单的管道将对图像进行随机旋转和缩放,并生成10,000个新样本。

主要功能

弹性变形

Augmentor可以对图像进行弹性变形,生成真实可行且保持标签的新样本:

弹性变形示例

透视变换

Augmentor提供12种不同类型的透视变换,包括倾斜、倾斜和扭曲等:

透视变换示例

保持大小的旋转

Augmentor的旋转操作默认会保持原图像的大小:

旋转示例

保持大小的剪切

剪切操作也会自动裁剪正确的区域,避免出现黑色空白:

剪切示例

随机擦除

随机擦除技术可以提高模型对遮挡的鲁棒性:

随机擦除示例

高级用法

多掩码/图像增强

Augmentor支持同时增强多个相关联的掩码图像:

多掩码增强

与Keras和PyTorch集成

Augmentor可以作为Keras和PyTorch的数据生成器使用:

# Keras示例
g = p.keras_generator(batch_size=128)
images, labels = next(g)

# PyTorch示例
import torchvision
transforms = torchvision.transforms.Compose([
    p.torch_transform(),
    torchvision.transforms.ToTensor(),
])