celisun/ImageCaptioning.pytorch

Image captioning codebase in pytorch(finetunable cnn in branch "with_finetune";diverse beam search can be found in 'dbs' branch; self-critical training is under my self-critical.pytorch repository.)

Jupyter Notebook