Skip to content

Image caption work with attention mechanism. Under Construction.

Notifications You must be signed in to change notification settings

liuchang8am/ImageSayWithAttention

Repository files navigation

Image caption work in lua.
On the basis of Torch7 framework.
Under Development.

Requirements

  1. torch7 http://torch.ch/docs/getting-started.html
  2. several packages besides default torch7:
    (1) dp: luarocks install dp
    (2) fs: luarocks install fs
    (3) rnn: luarocks install rnn
    (4) hdf5: luarocks install hdf5

Steps

1. Prepare your data

1) Run TorchPrepare.py

This file reads the raw sentences, and prepares them into json files to be further processed

2) Run prepo.py

This file converts the json files into h5 file, to be further read by lua script

TODO: 1)intergrate TorchPrepare.py and prepo.py into one single file, as preprocessing

About

Image caption work with attention mechanism. Under Construction.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages