Downloads · 30 days
0
efficietnet-lstm-image-captioning
Nanohana/efficietnet-lstm-image-captioning
efficietnet-lstm-image-captioning is a machine learning model from Nanohana. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository contains an image captioning system that is composed of:
Access
Public
Updated Nov 5, 2022
Repo size
28.1 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.gz28.1 MB · 100%
At a glance
- Access
- Public
- Created
- Nov 4, 2022
- Updated
- Nov 5, 2022
- SHA
- 329b6877
Model card
From the Hugging Face model README
image-captioning
This repository contains an image captioning system that is composed of:
- Pretrained EfficientNet-B0 in ImageNet
- Word Embedding with Flickr8k vocabulary
- 1 layer LSTM
It was trained for 100 epoches (CNN weights were frozen) and the vocabulary was built with words that appear at least 5 times in the Flickr8k dataset.
- Created
- Nov 4, 2022
- Updated
- Nov 5, 2022