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.
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}
}