Public | Automated Build

Last pushed: 3 years ago
Short Description
See the world
Full Description


See the world

Training your own RNN on the Flickr or Coco datasets

Fetch data

To download the datasets used for training, run


Then move the unpacked data to the data folder in the neuraltalk directory from within the docker container.

Train the RNN

Replacing flickr30k with coco or flickr8k as desired, run

python --dataset flickr30k

Using a pretrained RNN from Caffe Model Zoo

Classifying new images

Docker Pull Command
Source Repository