Skip to content

Latest commit

History

History
73 lines (57 loc) 路 3.2 KB

File metadata and controls

73 lines (57 loc) 路 3.2 KB

馃Auto Caption Generation for Images馃摳


Image Captioning is the process of generating textual description of an image. It uses both Natural Language Processing and Computer Vision to generate the captions. Deep Learning using CNNs-LSTMs can be used to solve this problem of generating a caption for a given image, hence called Image Captioning.

鈱汷utput that we get馃捇

There's a lot of biasing as well, since training data wasn't big enough!

馃枑锔廘ETS TAKE A QUICK DIVE INTO THIS BIT OF MAGIC!馃槆

Dataset:

  • Flickr 8k (containing 8k images),
  • Flickr 30k (containing 30k images),
  • MS COCO (containing 180k images), etc.

Point to Note:

Here I have used the Flickr8k dataset based on the availability of standard computational resources. This dataset is the best for 8GB RAM, and takes about 25mins/epoch training on a CPU. Flickr30k and MS COCO may need about 32GB-64GB RAM based on how it's processed. Consider using AWS EC2 workstation for the best and fastest output. Its paid tho馃槥!

General Architecture

Model Architecture(VGG16 + LSTMs)

We remove the last 2 layers of VGG16 and pass it to 馃憞

馃搳Data that we feed into the Network!馃搧


References and Bibliography:

Paper