GitHub - d685shvarts/TICN

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
.idea		.idea
.ipynb_checkpoints		.ipynb_checkpoints
__pycache__		__pycache__
.gitignore		.gitignore
Explore_Dataset.ipynb		Explore_Dataset.ipynb
Explore_LSTM.ipynb		Explore_LSTM.ipynb
LSTM.ipynb		LSTM.ipynb
LSTM.py		LSTM.py
README.txt		README.txt
RNN.py		RNN.py
TRAIN_RNN.ipynb		TRAIN_RNN.ipynb
TestImageIds.csv		TestImageIds.csv
TrainImageIds.csv		TrainImageIds.csv
Train_LSTM.ipynb		Train_LSTM.ipynb
[RNN_Trained_Embedding]info_over_epoch.txt		[RNN_Trained_Embedding]info_over_epoch.txt
data_loader.py		data_loader.py
evaluate_captions.py		evaluate_captions.py
get_datasets.ipynb		get_datasets.ipynb
vocab.py		vocab.py
wordArray.pkl		wordArray.pkl
wordDict.pkl		wordDict.pkl
wordStat.pkl		wordStat.pkl
word_dict.txt		word_dict.txt

Repository files navigation

Description of files:


data_loader.py - Create Pytorch Dataset and data loader for COCO dataset

evaluate_captions.py - Provides evaluation function to calculate BLEU1 and BLEU4 scores from true and predicted captions

get_datasets.ipynb - Python notebook to fetch COCO dataset from DSMLP cluster's root directory and place it in 'data' folder. Gets both images and annotations.


TestImageIds.csv - COCO dataset image ids for test set

TrainImageIds.csv - COCO dataset image ids for train set