ImageCaptioning_pytorch

Steps:

Clone git repository

git clone https://github.com/z0CoolCS/ImageCaptioning_pytorch.git
cd ImageCaptioning_pytorch

Create workspace

mkdir -p data/img/train
mkdir -p data/img/val
mkdir -p checkpts

Download JSON and JPG files

python -m download_data

Train the Image Caption Model

python -m main

Predict with trained model

python -m predict

Results