bhavanajain / show-and-tell

cs547 project on implementing image captioning (show and tell approach)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

show-and-tell

cs547 project on implementing image captioning (show and tell approach)

Things to try

  • Try Glove embedding instead of nn.Embedding
  • Increase encoder capacity: resnet50 -> resnet152
  • Try GRU instead of LSTM
  • Beam search with variable k
  • DataParallel + Distributed training

About

cs547 project on implementing image captioning (show and tell approach)


Languages

Language:Python 88.8%Language:Shell 11.2%