-
Notifications
You must be signed in to change notification settings - Fork 2
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
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
About
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.
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published