Skip to content

Image captioning project using the Flickr8k dataset. A custom encoder-decoder architecture was implemented based on ResNet-50 and LSTM, trained from scratch using PyTorch. The model takes an image as input and generates a descriptive caption. BLEU score was used for evaluation.

Notifications You must be signed in to change notification settings

Amir-Hofo/Image_Captioning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image_Captioning

Image captioning project using the Flickr8k dataset. A custom encoder-decoder architecture was implemented based on ResNet-50 and LSTM, trained from scratch using PyTorch. The model takes an image as input and generates a descriptive caption. BLEU score was used for evaluation.

Example output:
Target → "A BMX rider in a red and black outfit is jumping on his motorcycle."

IC

About

Image captioning project using the Flickr8k dataset. A custom encoder-decoder architecture was implemented based on ResNet-50 and LSTM, trained from scratch using PyTorch. The model takes an image as input and generates a descriptive caption. BLEU score was used for evaluation.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published