GitHub - xaviolo99/dual-transfer-caption-generation: This repository contains the notebooks used for training and evaluating a model which allows users to input images and get captions for the images back. It combines EfficientNet and GPT-2 pretrained models and a model to merge the outputs of the previous models.

xaviolo99 / dual-transfer-caption-generation Public

Notifications You must be signed in to change notification settings
Fork 2
Star 4

This repository contains the notebooks used for training and evaluating a model which allows users to input images and get captions for the images back. It combines EfficientNet and GPT-2 pretrained models and a model to merge the outputs of the previous models.

4 stars 2 forks Branches Tags Activity

Star

Notifications

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
code		code
extra_references.txt		extra_references.txt
final_project_description.pdf		final_project_description.pdf
report.md		report.md
report_xavier_reyes.pdf		report_xavier_reyes.pdf