Skip to content

kr1shnasomani/CaptionCraft

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CaptionCraft

The project is an image captioning model using DenseNet201 and LSTM, generating descriptive captions for images from the Flickr8k dataset. Utilized TensorFlow, Keras, and custom data generators for feature extraction and sequence prediction.

Execution Guide:

  1. Clone the repository:

    git clone https://github.com/kr1shnasomani/CaptionCraft.git
    cd CaptionCraft
    
  2. Install the dependencies:

    pip install -r requirements.txt
    
  3. On running all the cells it outputs a file named model.keras (it stores the trained model)

Accuracy & Loss Over Epochs:

image

image

Model Prediction:

image

About

Image Captioner using DenseNet201 and LSTM

Topics

Resources

Stars

Watchers

Forks