Project for CSC2548: Machine Learning for Computer Vision
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
coco_eval_kit
experiments
scripts
.gitignore
Readme.md
caption_net.py
coco_data_loader.py
evaluation_mode.py
forgetful_lstm_cell.py
main.py
training_mode.py
word_embedding.py

Readme.md

Experiments with RNNs for image captioning

Setup Project

Download COCO2014 training and validation set

Download VGG pretrained model:

wget https://download.pytorch.org/models/vgg16-397923af.pth

Install Spacy and download 'en_core_web_md' word vectors

Run script most_common_words_in_coco.py to generate index <-> word mapping