Skip to content

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

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages