/WB_color_augmenter

PyTorch Transforms wrapper for emulating white balance effects for color augmentation.

Primary LanguageJupyter NotebookMIT LicenseMIT

White-Balance Emulator for Color Augmentation

This is a PyTorch wrapper for the repository: https://github.com/mahmoudnafifi/WB_color_augmenter

See example.ipynb for a simple tutorial. Put the WB_Augmenter folder in your utils directory and import the WB_Augmenter.py module.

MIT License

Publication

If you use this code or our dataset, please cite our paper:

Mahmoud Afifi and Michael S. Brown. What Else Can Fool Deep Learning? Addressing Color Constancy Errors on Deep Neural Network Performance. International Conference on Computer Vision (ICCV), 2019.

@InProceedings{Afifi_2019_ICCV,
author = {Afifi, Mahmoud and Brown, Michael S.},
title = {What Else Can Fool Deep Learning? Addressing Color Constancy Errors on Deep Neural Network Performance},
booktitle = {The IEEE International Conference on Computer Vision (ICCV)},
month = {October},
year = {2019}
}