albumentations¶
albumentations is a fast image augmentation library and easy to use wrapper around other libraries.
Features¶
- Great fast augmentations based on highly-optimized OpenCV library.
- Super simple yet powerful interface for different tasks like (segmentation, detection, etc).
- Easy to customize.
- Easy to add other frameworks.
Project info¶
- GitHub repository: https://github.com/albumentations-team/albumentations
- GitHub repository with examples: https://github.com/albumentations-team/albumentations_examples
- License: MIT
Installation¶
You can use pip to install albumentations:
pip install albumentations
If you want to get the latest version of the code before it is released on PyPI you can install the library from GitHub:
pip install -U git+https://github.com/albumentations-team/albumentations
Demo¶
You can use this Google Colaboratory notebook to adjust image augmentation parameters and see the resulting images.
Contents:
- Examples
- Contributing
- To create a pull request:
- Augmentations overview
- API
- About probabilities.
- Writing tests
- A first test.
- Test parametrization and the @pytest.mark.parametrize decorator.
- Simplifying tests for functions that work with both images and masks by using helper functions.
- Simplifying tests for functions that work with both images and masks by using parametrization.
- Using fixtures.
- Simultaneous use of fixtures and parametrization.
- Hall of Fame
- Citations