nrtk-albumentations 2.2.1
Fork of Albumentations in direct support of NRTK (Natural Robustness Toolkit). Fast, flexible, and advanced augmentation library for deep learning, computer vision, and medical imaging. Offers a wide range of transformations for both 2D (images, masks, bboxes, keypoints) and 3D (volumes, volumetric masks, keypoints) data, with optimized performance and seamless integration into ML workflows.
License: MIT
Requires Python: <4.0,>=3.10
Keywords: 2D augmentation, 3D augmentation, aerial photography, anomaly detection, artificial intelligence, autonomous driving, bounding boxes, classification, computer vision, computer vision library, data augmentation, data preprocessing, data science, deep learning, deep learning library, depth estimation, face recognition, fast augmentation, image augmentation, image processing, image transformation, images, instance segmentation, keras, keypoint detection, keypoints, machine learning, machine learning tools, masks, medical imaging, microscopy, object counting, object detection, optimized performance, panoptic segmentation, pose estimation, python library, pytorch, quality inspection, real-time processing, robotics vision, satellite imagery, semantic segmentation, tensorflow, volumes, volumetric data, volumetric masks

# Albumentations (NRTK Fork)

[![CI](https://github.com/Kitware/nrtk-albumentations/workflows/CI/badge.svg)](https://github.com/Kitware/nrtk-albumentations/actions)
[![License: MIT](https://img.shields.io/badge/License-MIT-brightgreen.svg)](https://opensource.org/licenses/MIT)

This is a fork of [Albumentations](https://github.com/albumentations-team/albumentations) maintained by Kitware in direct support of [NRTK (Natural Robustness Toolkit)](https://github.com/Kitware/nrtk).

**Fork Information:**

- Last upstream version: Albumentations v2.0.8
- Forked: October 2025

## About

Albumentations is a Python library for fast and flexible image augmentation. This fork is maintained specifically for integration with NRTK and includes modifications to support NRTK's natural robustness evaluation workflows.

Image augmentation is used in deep learning and computer vision tasks to increase the quality of trained models by creating new training samples from existing data through various transformations.

## Key Features

- **Complete Computer Vision Support**: Works with all major CV tasks including classification, segmentation (semantic & instance), object detection, and pose estimation
- **Simple, Unified API**: One consistent interface for all data types - RGB/grayscale/multispectral images, masks, bounding boxes, and keypoints
- **Rich Augmentation Library**: 70+ high-quality augmentations to enhance your training data
- **Fast**: Optimized for production use with consistently high performance
- **Deep Learning Integration**: Works with PyTorch, TensorFlow, and other frameworks
- **3D Support**: Volumetric data transformations for medical imaging and other 3D applications

## Installation

This package is designed to be installed as a dependency of NRTK. For standalone installation, use:

```bash
pip install nrtk-albumentations (opencv-python|opencv-python-headless)
```

## A Simple Example

This package is intended for use with NRTK's [AlbumentationPerturber](https://nrtk.readthedocs.io/en/latest/_implementations/nrtk.impls.perturb_image.generic.albumentations_perturber.AlbumentationsPerturber.html#albumentationsperturber). For direct usage:

```python
import albumentations as A
import cv2

# Declare an augmentation pipeline
transform = A.Compose([
    A.RandomCrop(width=256, height=256),
    A.HorizontalFlip(p=0.5),
    A.RandomBrightnessContrast(p=0.2),
])

# Read an image with OpenCV and convert it to the RGB colorspace
image = cv2.imread("image.jpg")
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

# Augment an image
transformed = transform(image=image)
transformed_image = transformed["image"]
```

## Documentation

Albumentations documentation is [hosted by the original authors](https://albumentations.ai/docs/). An [offline copy](./docs/albumentations-documentation.pdf) of the documentation has been included with this repository, with the expectation that the hosted documentation may eventually become unavailable or diverge from the forked functionality. This offline copy was generated from [this commit hash](https://github.com/albumentations-team/albumentations-ai-docs/tree/624837ded1550ab528f5b77e4f91d37cf7982a27). Note that some external files and links may not be functional.

For usage within the NRTK ecosystem, please refer to [NRTK Documentation](https://nrtk.readthedocs.io/).

## Available Transformations

Albumentations provides 70+ transforms across several categories:

### Pixel-level Transforms

Transforms that modify pixel values without changing image geometry (masks, bboxes, keypoints remain unchanged):

- Color adjustments: `RandomBrightnessContrast`, `HueSaturationValue`, `ColorJitter`
- Noise addition: `GaussNoise`, `ISONoise`, `MultiplicativeNoise`
- Blur effects: `GaussianBlur`, `MotionBlur`, `MedianBlur`, `Defocus`
- Compression: `ImageCompression`
- And many more...

### Spatial-level Transforms

Transforms that modify image geometry (automatically applied to masks, bboxes, keypoints):

- Cropping: `RandomCrop`, `CenterCrop`, `RandomResizedCrop`
- Flipping: `HorizontalFlip`, `VerticalFlip`
- Rotation: `Rotate`, `RandomRotate90`, `SafeRotate`
- Resizing: `Resize`, `LongestMaxSize`, `SmallestMaxSize`
- Distortions: `ElasticTransform`, `GridDistortion`, `OpticalDistortion`
- And many more...

### 3D Transforms

Transforms for volumetric data (medical imaging, etc.):

- `RandomCrop3D`, `CenterCrop3D`
- `Pad3D`, `PadIfNeeded3D`
- `CoarseDropout3D`
- `CubicSymmetry`

For a complete list with detailed parameters, see the [transform reference](https://nrtk-albumentations.readthedocs.io/).

## Maintainer

**Kitware, Inc.** <nrtk@kitware.com>

## Original Authors

This library was originally created by:

- Vladimir I. Iglovikov
- Alexander Buslaev
- Alex Parinov
- Eugene Khvedchenya
- Mikhail Druzhinin

## Contributing

This fork is maintained specifically for NRTK integration. We are not accepting general contributions at this time. For issues or questions related to NRTK integration, please open an issue on the [NRTK repository](https://github.com/Kitware/nrtk/issues).

## License

This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

## Citation

If you use this library in your research, please consider citing the original Albumentations paper:

```bibtex
@Article{info11020125,
    AUTHOR = {Buslaev, Alexander and Iglovikov, Vladimir I. and Khvedchenya, Eugene and Parinov, Alex and Druzhinin, Mikhail and Kalinin, Alexandr A.},
    TITLE = {Albumentations: Fast and Flexible Image Augmentations},
    JOURNAL = {Information},
    VOLUME = {11},
    YEAR = {2020},
    NUMBER = {2},
    ARTICLE-NUMBER = {125},
    URL = {https://www.mdpi.com/2078-2489/11/2/125},
    ISSN = {2078-2489},
    DOI = {10.3390/info11020125}
}
```