Skip to content

Files

Latest commit

Apr 22, 2015
4ce9272 · Apr 22, 2015

History

History
This branch is up to date with imclab/neuraltalk:master.

imagernn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Nov 28, 2014
Nov 28, 2014
Nov 28, 2014
Nov 28, 2014
Nov 28, 2014
Nov 28, 2014
Jan 9, 2015
Apr 22, 2015
Nov 28, 2014

The code is organized as follows:

  • data_provider.py abstracts away the datasets and provides uniform API for the code.
  • utils.py is what it sounds like it is :)
  • solver.py: the solver class doesn't know anything about images or sentences, it gets a model and the gradients and performs a step update
  • generic_batch_generator.py handles batching across a batch of image/sentences that need to be forwarded through the networks. It calls the
  • lstm_generator.py, which is an implementation of the Google LSTM for generating images.
  • imagernn_utils.py contains some image-rnn specific utilities, such as evaluation function etc. These come in handy when we want to use some functionality across different scripts (e.g. driver and evaluator)
  • rnn_generator.py has a simple RNN implementation for now, an alternative to LSTM