albumentations是一个快速的图像增强库,用于机器学习任务。它支持各种类型的图像变换,包括但不限于旋转、平移、缩放、剪切、翻转、噪声注入、遮挡等。albumentations库可以与深度学习框架如PyTorch和TensorFlow很好地集成, 支持种类丰富的像素级变换,包括雨天、雾天、色域变换等等。
https://github.com/albumentations-team/albumentations
- 安装
pip install albumentations
- 导入库
import albumentations as A
- 创建变换组合:
transform = A.Compose(