Python图像处理库 – Albumentations,可用于深度学习中网络训练时的图片数据增强。Albumentations 图像数据增强库特点:
- 基于高度优化的 OpenCV 库实现图像快速数据增强;
- 针对不同图像任务,如分割,检测等,超级简单的 API 接口;
- 易于个性化定制;
- 易于添加到其它框架,比如 PyTorch;

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

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