This repository comprises the image sentiment analysis model training for the GANs & Roses project. It includes some utils, including bulk downloading the paintings in The Wikiart Emotions Dataset and some pre-processing to get the annotations fit our needs.
For the end-to-end, ready for deployment art to music pipeline, check out our deploy repository. If you are interested in geting to know more about the GANs & Roses project, you might enjoy reading our Medium article.
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.