albumentations库函数的简介
albumentations是图像数据增强库。基于高度优化的 OpenCV 库实现图像快速数据增强。针对不同图像任务,如分割,检测等,超级简单的 API 接口。易于个性化定制。易于添加到其它框架,比如 PyTorch。
GitHub官网:https://github.com/albumentations-team/albumentations
文档:https://albumentations.readthedocs.io/en/latest/api/augmentations.html
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.ipynbhttps://github.com/albu/albumentations/blob/master/notebooks/example.ipynb
Object detection - example_bboxes.ipynbhttps://github.com/albu/albumentations/blob/master/notebooks/example_bboxes.ipynb
Non-8-bit images - example_16_bit_tiff.ipynbhttps://github.com/albu/albumentations/blob/master/notebooks/example_16_bit_tiff.ipynb
Image segmentation example_kaggle_salt.ipynbhttps://github.com/albu/albumentations/blob/master/notebooks/example_kaggle_salt.ipynb
Keypoints example_keypoints.ipynbhttps://github.com/albu/albumentations/blob/master/notebooks/example_keypoints.ipynb