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

No packages published