/End-to-End-Image-Captioning

Primary LanguagePythonMIT LicenseMIT

Image-Captioning-Project

This project utilizes deep learning techniques to automatically generate descriptive captions for images. It also contains a self-hosted web server through Flask for users to upload their own images.

Features

  • Automatically generates captions for uploaded images.
  • Uses a pretrained deep learning model to understand image content.
  • Provides a seamless integration of computer vision and natural language processing.

Installation

  1. Clone this repository to your local machine using:

    git clone https://github.com/Asatheesh6561/End-To-End-Image-Captioning.git

  2. Navigate to the project directory:

    cd End-To-End-Image-Captioning

  3. Install the required dependencies:

    pip install -r requirements.txt

Usage

  1. Train the CNN-Transformer model by running this command:
python main.py
  1. Run the prediction program on your local machine hosted by Flask using this command:
python app.py