Skip to content

rezakarimzadeh/Image_captioning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image_captioning

Image captioning using RNN and CNN

cnn_for_captioning.ipynb -------> cnn model train for multi label classification

image_captioning.ipynb ----------->use cnn and rnn model and train a model for image captioning

use_cnn_labeling.ipynb------------> load cnn model and predict label (just open and change data path)

use_image_captioning_model.ipynb---------------> load image captioning model and use it for generate new captions (just open and change data path)

Note: decoder_model.h5 & cnn_multi_label.h5 links are as follow:

decoder_model.h5 link: https://drive.google.com/open?id=1Sduvt3jXkWm61SvBgt39qYQilpc6qRZk

cnn_multi_label.h5 link: https://drive.google.com/open?id=1FRM3eFZPmy4QGUolfTOOzusrFesHBH2b

About

Image captioning using RNN and CNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published