Skip to content

A data generator that considers both, the albumentations library and the classical Keras augmentation algorithm.

License

Notifications You must be signed in to change notification settings

mjkvaak/Augmentation_project

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

38 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Description

This small project compares the performance of the Keras classical augmentation algorithms and some configurations of the albumentations library.

Performance is tested in two different jupyter notebooks:

  • samples/FileDataGenerator_test --> compares the execution time to augment all the dataset.
  • samples/Neural_Network_test --> compares the performance of both approaches on the same dataset. Albumentations library results are quite promising (check the loss function graph and the execution time).

For this purpose, I've developed my own data generator, that read images, not from a directory but from a list of images and their corresponding labels. The FileDataGen class (at FileDataGenerator.py) considers both, Keras classical augmentation algorithm and the albumentations library.

New albumentations configurations can be implemented by modifying the augmentation_clss (at augment.py).

The albumentations library has been developed by Buslaev Alexander, Alexander Parinov, Vladimir Iglovikov.




BibTex:

@misc{Freire18_aug,
author = {Freire-Obreg'on, D.},
title = {Augmentation Project},
year = {2018},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/davidfreire/Augmentation_project}},
doi= {10.5281/zenodo.1438467}
}

About

A data generator that considers both, the albumentations library and the classical Keras augmentation algorithm.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%