ram-ek / Image-Captioning

Image Captioning

Overview

• Image captioning model which predicts the caption of a given image.
• The dataset used is COCO 2014.
• Faiss Library is used for efficient similarity search and clustering.

Execution

• Run this notebook in jupyter and run all cells.
• It will also download the COCO dataset.

Note

• The notebook accesses the COCO dataset through mounted google drive, change it to point to where the COCO dataset is available.