Merge branch 'master' of https://github.com/seung-kim/pmn_demo

README.md

@@ -1,5 +1,7 @@
 # pmn_demo
-Code for running trained model from Visual Reasoning by Progressive Module Networks (ICLR19)
+This repository contains code for running trained model from Visual Reasoning by Progressive Module Networks (ICLR19)
+
+Paper: https://arxiv.org/abs/1806.02453 (Arxiv), https://openreview.net/forum?id=B1fpDsAqt7 (OpenReview)
 
 Dependencies:
 pytorch 0.3, python 2.7
@@ -10,6 +12,15 @@ To run the code:
 3. Run python generate_html.py full-path-to-saved-outputs(default: ./test)
 4. Find result.html in full-path-to-saved-outputs(default: ./test)
 
+Example output:
+
+<img src="demo_output/im1.png" width="300">
+<img src="demo_output/im2.png" width="500">
+<img src="demo_output/im3.png" width="500">
+<img src="demo_output/im4.png" width="500">
+<img src="demo_output/im5.png" width="500">
+<img src="demo_output/im6.png" width="500">
+<img src="demo_output/im7.png" width="500">
 
 (Optionally):
 To run on your own data, you need to extract bounding boxes using the model from https://github.com/peteanderson80/bottom-up-attention