Skip to content

Latest commit

 

History

History
15 lines (9 loc) · 691 Bytes

README.md

File metadata and controls

15 lines (9 loc) · 691 Bytes

Skeleton-key

The implementation of the model in paper "Skeleton Key: Image Captioning by Skeleton-Attribute Decomposition"

Prerequisite

The model uses tensorflow, and the preprocessing of the captions requires Stanford NLP Core and you need to download COCO dataset first.

Dataset preprocessing

Use create_data.py to create the skeleton-attribute dataset from COCO.

Test

Download the pre-trained model at Drive, and put the model under ./model Use run_inference.py to test the model on the 5000-split test set.