Update README.md

uvavision · Jul 26, 2017 · d3f616f · d3f616f
1 parent 421a4c7
commit d3f616f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,7 @@
 
 # OBJ2TEXT
+:point_right: Checkout [the Pytorch implementation](https://github.com/xuwangyin/pytorch-tutorial/tree/master/tutorials/03-advanced/image_captioning).
+
 OBJ2TEXT is a sequence-to-sequence model that takes object layout (sequence of pairs of object names and locations) as input and generates a natural language description of the input layout. This repository contains code for running all the experiments in the paper [OBJ2TEXT: Generating Visually Descriptive Language from Object Layouts](https://arxiv.org/abs/1707.07102). You can train different models by specifying different values for the option `-generate_from`:
  * `objclass`, generate caption from a sequence of object classes (the set of object names with all duplicates removed)
  * `objname`, generate caption from a sequence of object names