Skip to content

An ML model which tries to predict a caption by training the image

Notifications You must be signed in to change notification settings

Sathishkumar1012/Image-Captioning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 

Repository files navigation

Image-Captioning

In this project, our model tries to predict a caption based on the image given. We used flickr8k dataset for our training and testing purposes. We start with predicting the objects inside an image using CNN. We then use RNN to predict a caption for that image.

The folder image captioning.pynb contains the code.The screenshots contains the output of some trained images.

Tools: CNN,RNN,transfer learning,resnet-50

About

An ML model which tries to predict a caption by training the image

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published