目录
albumentations库函数的简介
albumentations是图像数据增强库。基于高度优化的 OpenCV 库实现图像快速数据增强。针对不同图像任务,如分割,检测等,超级简单的 API 接口。易于个性化定制。易于添加到其它框架,比如 PyTorch。
文档:Augmentations (albumentations.augmentations) — albumentations 1.1.0 documentation
1、albumentations库特点
- 在大多数转换中,这个库比其他库更快。
- 基于numpy, OpenCV, imgaug从每一个选择最好的。
- 简单、灵活的API,允许库在任何计算机视觉管道中使用。
- 大型的、不同的转换集。
- 很容易扩展这个库来包装其他库。
- 易于扩展到其他任务。
- 支持转换的图像,面具,关键点和包围框。
- 支持python 2.7 -3.7
- 易于与PyTorch集成。
- 容易从火炬传递。
- 曾在Kaggle、topcoder、CVPR、MICCAI等许多DL竞赛中获得过冠军。
- 由Kaggle Masters撰写。
albumentations库函数的安装
pip install albumentations
pip install --user albumentations
albumentations库函数的使用方法
1、经典案例
Classification - example.ipynb
Object detection - example_bboxes.ipynb
Non-8-bit images - example_16_bit_tiff.ipynb
Image segmentationexample_kaggle_salt.ipynb
Keypointsexample_keypoints.ipynb
Custom targetsexample_multi_target.ipynb
Weather transformsexample_weather_transforms.ipynb
Serializationserialization.ipynb
Replay/Deterministic modereplay.ipynb