wujian752 / e2e-glstm-resnet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

This repository includes the code for end-to-end gLSTM and sentence-conditional semantic attention, as appeared in the paper "Image Caption Generation with Text-Conditional Semantic Attention". train_new.lua is the main file for e2e-gLSTM and train_sc.lua is the main file for sentence-conditional semantic attention. The implementation is based on Neuraltalk2 (https://github.com/karpathy/neuraltalk2). Please follow the instructions on Neuraltalk2 to run the code. Contact me if you have any trouble running the code (http://luoweizhou.net/contact.html). Please cite this paper (http://arxiv.org/abs/1606.04621) if you are using the code.

About


Languages

Language:Jupyter Notebook 50.8%Language:Lua 45.1%Language:Python 3.4%Language:HTML 0.5%Language:Shell 0.2%