smittal6 / keras-captioning

Neural Image Captioning, Done as course project for CS771 @ IITK

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Usage:

python3 train.py Path of the folder is files/models/

Input format:

Each caption (with start and end tokens) is converted into a one-hot matrix of shape MAX_SEQUENCE_LENGTH x VOCAB_SIZE. Output is this matrix rotated one row upwards, so first word has second word corresponding to it.

About

Neural Image Captioning, Done as course project for CS771 @ IITK


Languages

Language:Python 100.0%