Image classification for everyone.
APACHE-2.0 License
Image ATM is a one-click tool that automates the workflow of a typical image classification pipeline in an opinionated way, this includes:
Read the documentation at: https://idealo.github.io/imageatm/
Image ATM is compatible with Python 3.6 and is distributed under the Apache 2.0 license.
There are two ways to install Image ATM:
pip install imageatm
git clone https://github.com/idealo/imageatm.git
cd imageatm
python setup.py install
Run this in your terminal
imageatm pipeline config/config_file.yml
Run the data preparation:
from imageatm.components import DataPrep
dp = DataPrep(
samples_file = 'sample_configfile.json',
image_dir = 'sample_dataset/',
job_dir='sample_jobdir/'
)
dp.run(resize=True)
Run the training:
from imageatm.components import Training
trainer = Training(image_dir=dp.image_dir, job_dir=dp.job_dir)
trainer.run()
Run the evaluation:
from imageatm.components import Evaluation
evaluator = Evaluation(image_dir=dp.image_dir, job_dir=dp.job_dir)
evaluator.run()
Test execution is triggered by these commands:
pip install -e ".[tests, docs]"
pytest -vs --cov=imageatm --show-capture=no --disable-pytest-warnings tests/
The following pretrained CNNs from Keras can be used for transfer learning in Image-ATM:
Training is split into two phases, at first only the last dense layer gets trained, and then all layers are trained.
For each phase the learning rate is reduced after a patience period if no improvement in validation accuracy has been observed. The patience period depends on the average number of samples per class (n_per_class):
Training is stopped early after a patience period that is three times the learning rate patience to allow for two learning rate adjustments before stopping training.
We welcome all kinds of contributions. See the Contribution guide for more details.
To bump up the version, use
bumpversion {part} setup.py
Please cite Image ATM in your publications if this is useful for your research. Here is an example BibTeX entry:
@misc{idealods2019imageatm,
title={Image ATM},
author={Christopher Lennan and Malgorzata Adamczyk and Gunar Maiwald and Dat Tran},
year={2019},
howpublished={\url{https://github.com/idealo/imageatm}},
}
See LICENSE for details.