Skip to content

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

Notifications You must be signed in to change notification settings

feiyu1990/Skeleton-key

Repository files navigation

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages