This project implements a RNN using features from a Resnet-18 to predict the theme of a photo album Example of predicted themes are :
- Birthday
- Protest
- Zoo
We use images from the The Curation of Flickr Events Dataset (CUFED) dataset
Here are some example of pics of the albums :
The project can easily be run with the latest docker image of pytorch :
sudo nvidia-docker run -v $$$$:/code --rm -ti --ipc=host pytorch/pytorch:latest
Once in the docker, run the install of packages :
pip install scikit-image tensorbaordx
To run the training, please launch :
python run_training