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:

Model Prediction: