nrtk-albumentations 2.1.0__tar.gz

nrtk_albumentations-2.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2017 Vladimir Iglovikov, Alexander Buslaev, Alexander Parinov,
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

nrtk_albumentations-2.1.0/PKG-INFO

@@ -0,0 +1,196 @@
+Metadata-Version: 2.4
+Name: nrtk-albumentations
+Version: 2.1.0
+Summary: 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
+License-File: LICENSE
+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
+Author: Vladimir Iglovikov
+Maintainer: Kitware, Inc.
+Maintainer-email: nrtk@kitware.com
+Requires-Python: >=3.9
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Healthcare Industry
+Classifier: Intended Audience :: Information Technology
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: Astronomy
+Classifier: Topic :: Scientific/Engineering :: Atmospheric Science
+Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Classifier: Topic :: Scientific/Engineering :: Physics
+Classifier: Topic :: Scientific/Engineering :: Visualization
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Provides-Extra: hub
+Provides-Extra: pytorch
+Provides-Extra: text
+Requires-Dist: PyYAML
+Requires-Dist: albucore (==0.0.28)
+Requires-Dist: eval-type-backport ; python_version < "3.10"
+Requires-Dist: numpy (>=1.24.4)
+Requires-Dist: opencv-python-headless (>=4.9.0.80)
+Requires-Dist: pydantic (>=2.9.2)
+Requires-Dist: scipy (>=1.10.0)
+Requires-Dist: typing-extensions (>=4.9.0) ; python_version < "3.10"
+Project-URL: Homepage, https://nrtk.readthedocs.io/
+Project-URL: Repository, https://github.com/Kitware/nrtk-albumentations
+Description-Content-Type: text/markdown
+
+# 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
+```
+
+## 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
+
+For detailed documentation on available transforms and usage patterns, see:
+
+- [Albumentations (NRTK Fork) Documentation](https://nrtk-albumentations.readthedocs.io/)
+- [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}
+}
+```
+

nrtk_albumentations-2.1.0/README.md

@@ -0,0 +1,142 @@
+# 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
+```
+
+## 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
+
+For detailed documentation on available transforms and usage patterns, see:
+
+- [Albumentations (NRTK Fork) Documentation](https://nrtk-albumentations.readthedocs.io/)
+- [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}
+}
+```

nrtk_albumentations-2.1.0/albumentations/__init__.py

@@ -0,0 +1,21 @@
+from importlib.metadata import metadata
+
+try:
+    _metadata = metadata("albumentations")
+    __version__ = _metadata["Version"]
+    __author__ = _metadata["Author"]
+    __maintainer__ = _metadata["Maintainer"]
+except Exception:  # noqa: BLE001
+    __version__ = "unknown"
+    __author__ = "Vladimir Iglovikov"
+    __maintainer__ = "Kitware, Inc."
+
+from contextlib import suppress
+
+from .augmentations import *
+from .core.composition import *
+from .core.serialization import *
+from .core.transforms_interface import *
+
+with suppress(ImportError):
+    from .pytorch import *

nrtk_albumentations-2.1.0/albumentations/augmentations/__init__.py

@@ -0,0 +1,23 @@
+from .blur.transforms import *
+from .crops.transforms import *
+from .dropout.channel_dropout import *
+from .dropout.coarse_dropout import *
+from .dropout.grid_dropout import *
+from .dropout.mask_dropout import *
+from .dropout.transforms import *
+from .dropout.xy_masking import *
+from .geometric.distortion import *
+from .geometric.flip import *
+from .geometric.pad import *
+from .geometric.resize import *
+from .geometric.rotate import *
+from .geometric.transforms import *
+from .mixing.domain_adaptation import *
+from .mixing.transforms import *
+from .other.lambda_transform import *
+from .other.type_transform import *
+from .pixel.transforms import *
+from .spectrogram.transform import *
+from .text.transforms import *
+from .transforms3d.transforms import *
+from .utils import *