Skip to content

Generates a caption for the given image and analyzes the generated caption's sentiment.

Notifications You must be signed in to change notification settings

a1m0/caption-generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Caption Generator

Generates a caption for the given image and analyzes the generated caption's sentiment.

Overview

Part 1: Generate Objective Captions

Using flickr8k and COCO Datasets and combination of transfer learning and custom NLP neural networks, created two different models for generating Objective Captions.

Part 1.5: Generate Captions with Sentiment

Using COCO and SentiCap Datasets and combination of transfer learning and a custom NLP neural network, created a model for generating Captions with Sentiment.

Part 2: Analyze Generated Caption's Sentiment

Created/Used 4 different models for sentiment analysis of captions:

  1. Texblob
  2. Vader
  3. Hugging Face
  4. Custom LSTM Neural Network

About

Generates a caption for the given image and analyzes the generated caption's sentiment.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published