Notebooks

If you found errors in the notebooks, please create:

1. Pull request with the fix or
2. Submit an issue to the main repository of Albumentations

• example.ipynb. Defining a simple augmentation pipeline for image augmentation.
• example_bboxes.ipynb. Using Albumentations to augment bounding boxes for object detection tasks.
• example_bboxes2.ipynb. How to use Albumentations for detection tasks if you need to keep all bounding boxes.
• example_kaggle_salt.ipynb. Using Albumentations for a semantic segmentation task.
• example_keypoints.ipynb. Using Albumentations to augment keypoints.
• face_landmarks_tutorial.ipynb. End-to-end tutorial on training a facial landmarks detection model with PyTorch and Albumentations.
• example_multi_target.ipynb. Applying the same augmentation with the same parameters to multiple images, masks, bounding boxes, or keypoints.
• example_weather_transforms.ipynb. Weather augmentations in Albumentations.
• migrating_from_torchvision_to_albumentations.ipynb. Migrating from torchvision to Albumentations.
• replay.ipynb. Debugging an augmentation pipeline with ReplayCompose.
• serialization.ipynb. How to save and load parameters of an augmentation pipeline.
• showcase.ipynb. Showcase. Cool augmentation examples on diverse set of images from various real-world tasks.
• pytorch_classification.ipynb. PyTorch and Albumentations for image classification.
• keras_cats_dogs_classification.ipynb. Keras/TensorFlow and Albumentations for image classification with automatic device detection (CUDA/MPS/CPU).
• keras_pretrained_segmentation.ipynb. Semantic segmentation with pretrained U-Net in Keras 3.x using Albumentations. Transfer learning from ImageNet (ResNet50, MobileNetV2, VGG16).
• pytorch_semantic_segmentation.ipynb. PyTorch and Albumentations for semantic segmentation.
• example_ultralytics.ipynb. Using custom Albumentations augmentations with Ultralytics YOLO models (by Andrei Alkhouski).
• xy_transform.ipynb. How to apply XYMasking.
• example_chromatic_aberration.ipynb. How to apply ChromaticAberration.
• example_d4.ipynb. How to apply D4 transform.
• example_gridshuffle.ipynb. How to apply RandomGridShuffle transform.
• example_documents.ipynb. How to apply Morphological, both erosion and dilation.
• example_hfhub.ipynb. How to load and save augmentation pipeline to HuggungFace Hub.
• example_OverlayElements.ipynb. How to apply OverlayElements to update text on the image
• example_textimage.ipynb. How to add text to the image
• example_mosaic.ipynb. How to create mosaic augmentations

Usage examples

For detailed examples, see notebooks.