Skip to content

ZhouyaoXie/image-caption-generation-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

image-caption-generation-pytorch

Final project for CMU 10701 Introduction to Machine Learning for PhDs.

The notebook implements a encoder-decoder model with scheduled sampling and beam search. The model is trained on the Microsoft Common Objects in Context (MSCOCO) data. For implementation details, please check out the notebook and the report.

About

An image caption generation model using Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published