autoalbument

AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/

Showing:

Popularity

Downloads/wk

0

GitHub Stars

127

Maintenance

Last Commit

8mos ago

Contributors

1

Package

Dependencies

12

License

MIT

Categories

Readme

AutoAlbument

AutoAlbument is an AutoML tool that learns image augmentation policies from data using the Faster AutoAugment algorithm. It relieves the user from the burden of manually selecting augmentations and tuning their parameters. AutoAlbument provides a complete ready-to-use configuration for an augmentation pipeline.

The library supports image classification and semantic segmentation tasks. You can use Albumentations to utilize policies discovered by AutoAlbument in your computer vision pipelines.

The documentation is available at https://albumentations.ai/docs/autoalbument/

Benchmarks

Here is a comparison between a baseline augmentation strategy and an augmentation policy discovered by AutoAlbument for different classification and semantic segmentation tasks. You can read more about these benchmarks in the autoalbument-benchmarks repository.

Classification

DatasetBaseline Top-1 AccuracyAutoAlbument Top-1 Accuracy
CIFAR1091.7996.02
SVHN98.3198.48
ImageNet73.2775.17

Semantic segmentation

DatasetBaseline mIOUAutoAlbument mIOU
Pascal VOC73.3475.55
Cityscapes79.4779.92

Installation

AutoAlbument requires Python 3.6 or higher. To install the latest stable version from PyPI:

pip install -U autoalbument

How to use AutoAlbument

How to use AutoAlbument

  1. You need to create a configuration file with AutoAlbument parameters and a Python file that implements a custom PyTorch Dataset for your data. Next, you need to pass those files to AutoAlbument.
  2. AutoAlbument will use Generative Adversarial Network to discover augmentation policies and then create a file containing those policies.
  3. Finally, you can use Albumentations to load augmentation policies from the file and utilize them in your computer vision pipelines.

You can read the detailed description of all steps at https://albumentations.ai/docs/autoalbument/how_to_use/

Examples

The examples directory contains example configs for different tasks and datasets:

Classification

Semantic segmentation

To run the search with an example config:

autoalbument-search --config-dir </path/to/directory_with_dataset.py_and_search.yaml>

Rate & Review

Great Documentation0
Easy to Use0
Performant0
Highly Customizable0
Bleeding Edge0
Responsive Maintainers0
Poor Documentation0
Hard to Use0
Slow0
Buggy0
Abandoned0
Unwelcoming Community0
100
No reviews found
Be the first to rate

Alternatives

No alternatives found

Tutorials

No tutorials found
Add a tutorial