Welcome to Albumentations documentation¶
Albumentations is a fast and flexible image augmentation library. The library is widely used in industry, deep learning research, machine learning competitions, and open source projects. Albumentations is written in Python, and it is licensed under the MIT license. The source code is available at https://github.com/albumentations-team/albumentations.
If you are new to image augmentation, start with articles in the "Introduction to image augmentation" section. They describe what image augmentation is, how it can boost deep neural networks' performance, and why you should use Albumentations.
Articles in the "Getting started with Albumentations" section show how you can use the library for different computer vision tasks: image classification, semantic segmentation, instance segmentation, and object detection, keypoint detection.
The "Examples" section contains Jupyter Notebooks that demonstrate how to use various features of Albumentations. Each notebook includes a link to Google Colab, where you can run the code by yourself.
"API Reference" contains the description of Albumentations' methods and classes.
Introduction to image augmentation¶
- What is image augmentation and how it can improve the performance of deep neural networks
- Why you need a dedicated library for image augmentation
- Why Albumentations
Getting started with Albumentations¶
- Installation
- Frequently Asked Questions
- Image augmentation for classification
- Mask augmentation for segmentation
- Bounding boxes augmentation for object detection
- Keypoints augmentation
- Simultaneous augmentation of multiple targets: masks, bounding boxes, keypoints
- A list of transforms and their supported targets
- Setting probabilities for transforms in an augmentation pipeline
Integrations¶
Examples¶
- Defining a simple augmentation pipeline for image augmentation
- Using Albumentations to augment bounding boxes for object detection tasks
- How to use Albumentations for detection tasks if you need to keep all bounding boxes
- Using Albumentations for a semantic segmentation task
- Using Albumentations to augment keypoints
- Applying the same augmentation with the same parameters to multiple images, masks, bounding boxes, or keypoints
- Weather augmentations in Albumentations
- Example of applying XYMasking transform
- Example of applying MixUp transform
- Example of applying ChromaticAberration transform
- Example of applying Morphological transform
- Example of applying D4 transform
- Example of applying RandomGridShuffle transform
- Example of applying OverlayElements transform
- Example of applying TextImage transform
- Migrating from torchvision to Albumentations
- Debugging an augmentation pipeline with ReplayCompose
- How to save and load parameters of an augmentation pipeline
- Showcase. Cool augmentation examples on diverse set of images from various real-world tasks.
- How to save and load transforms to HuggingFace Hub.
Examples of how to use Albumentations with different deep learning frameworks¶
- PyTorch and Albumentations for image classification
- PyTorch and Albumentations for semantic segmentation
- Using Albumentations with Tensorflow
External resources¶
- Blog posts, podcasts, talks, and videos about Albumentations
- Books that mention Albumentations
- Online courses that cover Albumentations