by Pranav Nedunghat (pn2187), Suraj Kiron Nair (skn8716), Ritvik Nair (pn2187)
Install requirements
pip install -r requirements.txt
Setup resource folder and download COCO dataset
./setup.sh
Create hdf5 files
python hdf5generator.py
Train the model: You can Tune hyperparameters in config.py
python train.py
Test the model
python test.py -W <best weight folder>
To perform inference
python inference.py