Skip to content

Latest commit

 

History

History
83 lines (51 loc) · 2.06 KB

README.md

File metadata and controls

83 lines (51 loc) · 2.06 KB
Raw

DeepCaptioning

Yet another implementation of im2txt (show and tell: A Neural Image Caption Generator).

Description

Image captioning aims to automatically generate natural language descriptions of images and im2txt is the model of image captioning developed by Google.

DeepCaptioning provides training and evaluation environments for creating an im2txt model.

Features

  • InceptionV3 encoder
  • LSTM decoder
  • Greedy and beam search decoding

Results on COCO dataset

result6 result7

Results on flickr8k dataset

result1 result2 result3

How to use

To generate a caption you will need to follow these steps.

Prepare the training data

DeepCaptioning uses the Flickr8k dataset. Please download the dataset from this link. Create the directory with the name of 'flickr8k' containing the dataset.

First of all, you need to pre-process caption texts which are tokenize, converting to lower case, removing puctuation, and applying some other filters.

$ python caption_preproc.py

The script below creates the captions with special start and end token, which are ground truth captions.

$ python gen_train_captions.py

To generate image features DeepCaptioning uses the pretrained InceptionV3 model. This script generates image features of train and test set and save as pkl.

$ python gen_image_feats.py

Finally generate the vocabulary from the training captions.

$ python word_embeddings.py

Train the model

Run the training script.

$ python train.py

If you train the model for 20 epochs(the loss is about 2.5), you will get the results like above.

Inference

Run the inference script.

$ python inference.py

This inference script selects the target image at random from the test image list.

License

MIT