arvindmn01/Caption_generation

Caption Generation Using Deep learning

I trained Deep learning model using CNN and LSTM to generate Caption on given Image.

User can Upload Image, Model will return the Caption.

I used Google Colab for model training. the link of Colab file is given below https://colab.research.google.com/drive/1iicqfTWMxKLrrQROXzrEGExQHTtS-o3D?usp=sharing

To star with project. First install the required python libraries using the below command

pip install -r requirements.txt

then simpy run the python file. it will create web app where you can generate caption for your image

Screenshots of Flask App.

flask Main page

Page after Caption generated By Model.