Skip to content

enigmarikki/Image-Captioning-Model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image-Captioning-Model

Unique Features of the Model are:


-> The model is capable on running on low-end GPUs such as the GTX 1050 with 4GB GPU Memory and 8GB RAM
-> The learning of the model is Steep
-> The Model was based on Images and captions scrapped from the E-Commerce Websites

Description of the Convolution Neural Network

image

Description of the Recurrent Neural Network model

image
The Picture above summarises the RNN model that was used.

The Features of each image was extracted through the VGG16 model with the last softmax layer popped out and connected to a RNN layer.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages