Skip to content
The implementation of the model in paper "Skeleton Key: Image Captioning by Skeleton-Attribute Decomposition"
Branch: master
Clone or download

README.md

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.

You can’t perform that action at this time.