/image-cap

Image Captioning using LSTM

Primary LanguageJupyter Notebook

Repository for the project

https://github.com/anthea97/image-cap

Team members

  • Hima Sai Kiran Prudhivi HXP220011
  • Anthea Abreo AXA210122
  • Naga Sri Harsha Vadrevu NXV210002
  • Lakshmi Harshitha Kamma HLK220000

Dataset

Install libraries

pip install pyspark
pip install findspark
pip install tensorflow

Start pyspark

pyspark

Results

1 epoch
BLEU-1: 0.476257
BLEU-2: 0.294542
start a dog dog end
start a young girl in a and in a end
start a man in a snow end
5 epochs
BLEU-1: 0.631810
BLEU-2: 0.444975
start two dogs are playing with a ball end
start a girl in a yellow dress is playing in the grass end
start a skier in a red jacket is skiing in the snow end
20 epochs
BLEU-1: 0.535163
BLEU-2: 0.303571
start two dogs play with on the sidewalk each other end
start two children are sitting in the below the rainbow end
start four people standing on snow in costumes end
100 epochs
BLEU-1: 0.934329
BLEU-2: 0.917832
black dog and white dog with brown spots are staring at each other in the street
little girl is sitting in front of large painted rainbow
skier looks at framed pictures in the snow next to trees