shourya1997 / image-caption-generator

Experiments with Neural Image captioning in Keras based on Show and Tell.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Image Captioning Generator

Experiments with Neural Image captioning in Keras based on Show and Tell.

Python Environment

  1. Python3
  2. Keras 2.1.5 or Higher
  3. Tensorflow 1.5 or Higher
  4. scikit-learn
  5. Pandas
  6. NumPy
  7. Matplotlib

How to Setup a Python Environment for Machine Learning and Deep Learning with Anaconda

How To Develop and Evaluate Large Deep Learning Models with Keras on Amazon Web Services

Dataset

Flickr8K dataset

  1. Flickr8k_Dataset.zip (1 Gigabyte) An archive of all photographs.
  2. Flickr8k_text.zip (2.2 Megabytes) An archive of all text descriptions for photographs.

Evaluation Metric

BLEU scores. For reference Where to put the Image in an Image Caption Generator

Model

Model Architecture

Model Structure

Model Structure

Experimentation Results

Train

Test

About

Experiments with Neural Image captioning in Keras based on Show and Tell.


Languages

Language:Jupyter Notebook 100.0%