Skip to content

Describe images with caption using machine learning and Pytorch framework

Notifications You must be signed in to change notification settings

DanBeverley/Image-Captioning-with-Pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Image-Captioning-with-Pytorch

Describe images with caption using machine learning and Pytorch framework

  • I used the Flickr8k dataset on Kaggle and trained the program on Google Colab with 10 epochs , the program perform quite well . Known issue is sometimes the description fumbling words image

About

Describe images with caption using machine learning and Pytorch framework

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages