Py之albumentations：albumentations库函数的简介、安装、使用方法之详细攻略

Py之albumentations：albumentations库函数的简介、安装、使用方法之详细攻略

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.ipynb

Object detection - example_bboxes.ipynb

Non-8-bit images - example_16_bit_tiff.ipynb

Image segmentation example_kaggle_salt.ipynb

Keypoints example_keypoints.ipynb

Custom targets example_multi_target.ipynb

Weather transforms example_weather_transforms.ipynb

Serialization serialization.ipynb

Replay/Deterministic mode replay.ipynb

赞 (0)

相关推荐