Metadata-Version: 2.1
Name: autoalbument
Version: 0.1.1
Summary: AutoML for image augmentation
Home-page: https://github.com/albumentations-team/autoalbument
Author: Alex Parinov, Vladimir Iglovikov, Eugene Khvedchenya, Druzhinin Mikhail, Buslaev Alexander
License: MIT
Platform: UNKNOWN
Classifier: Development Status :: 4 - Beta
Classifier: License :: OSI Approved :: MIT License
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.6
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Description-Content-Type: text/markdown
Requires-Dist: albumentations (>=0.5.1)
Requires-Dist: torch (>=1.6.0)
Requires-Dist: hydra-core (>=1.0)
Requires-Dist: timm (==0.1.20)
Requires-Dist: segmentation-models-pytorch
Requires-Dist: tqdm
Requires-Dist: click
Requires-Dist: colorama
Requires-Dist: tensorboard
Provides-Extra: tests
Requires-Dist: pytest ; extra == 'tests'

# AutoAlbument

AutoAlbument is an AutoML tool that learns image augmentation policies from data using the [Faster AutoAugment algorithm](https://arxiv.org/abs/1911.06987). 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](https://github.com/albumentations-team/albumentations) to utilize policies discovered by AutoAlbument in your computer vision pipelines.

The documentation is available at [https://albumentations.ai/docs/autoalbument/](https://albumentations.ai/docs/autoalbument/)

## 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](https://albumentations.ai/docs/images/autoalbument/how_to_use/autoalbument_usage.png)

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](https://github.com/albumentations-team/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/](https://albumentations.ai/docs/autoalbument/how_to_use/)


## Examples
The [`examples`](https://github.com/albumentations-team/autoalbument/tree/master/examples) directory contains example configs for different tasks and datasets:
- [Image classification on the CIFAR10 dataset](https://github.com/albumentations-team/autoalbument/tree/master/examples/cifar10).
- [Semantic segmentation on the Pascal VOC dataset](https://github.com/albumentations-team/autoalbument/tree/master/examples/pascal_voc).

To run the search with an example config:
```
autoalbument-search --config-dir </path/to/directory_with_dataset.py_and_search.yaml>
```


