Skip to content

pratikpv/image-captioning-through-rl

 
 

Repository files navigation

Image captioning using Deep Reinforcement Learning

  • Implementation of Zhou, R., Wang, X., Zhang, N., Lv, X., & Li-Jia, L. (2017). Deep Reinforcement Learning-based Image Captioning with Embedding Reward. ArXiv.org, Apr 12, 2017.
  • We used Advantage Actor Critic (A2C) Model to train the Policy network and the value network.
  • Also experimented with curriculum learning-based approaches.
  • Project completed as part of Reinforcement Learning class at SJSU, Spring 2020
  • Presentation

Sample output

image 1 image 2

About

Image captioning using Deep Reinforcement Learning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%