图像数据增强库 – Albumentations

Python图像处理库 – Albumentations,可用于深度学习中网络训练时的图片数据增强。Albumentations 图像数据增强库特点:

  • 基于高度优化的 OpenCV 库实现图像快速数据增强;
  • 针对不同图像任务,如分割,检测等,超级简单的 API 接口;
  • 易于个性化定制;
  • 易于添加到其它框架,比如 PyTorch;
Albumentations 颜色增强 - light
Albumentations 颜色增强 – light

以下是对 ImageNet Validation Set 中的前 2,000 张图片进行处理,采用 Intel Core i7-7800X CPU,不同数据增强库的处理速度对比(以秒为单位,时间越少越好) ——

不同数据增强库的处理速度对比
不同数据增强库的处理速度对比

这里有一篇比较完整的入门教程

GitHub地址项目 GitHub 地址