Skip to content

zlsh80826/image-caption-tf

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Caption Tensorflow

Requirements

Here are some required libraries.

General

  • python >= 3.6
  • cuda >= 10.0 (or base on your tensorflow version)

Python

  • please refer requirements.txt

Reproduce from scratch

Download the data

cd data
sh download.sh

Redistribute the data (Competition required)

python split.py

Generate the image features

We use the NASNet model pretrained by Keras to get the image features. This step may took over one hour.

python nasnet.py

Create tensorflow records

cd ../script
python create_tfrecord.py

Train

python train.py

Evaluate on validation set

python inference.py

Performance

CIDEr-D
Single Model 0.944
Ensemble Model 0.955