Model: tarekziade/vit-base-patch16-224-distilgpt2
Author: tarekziade
Pipeline: image-to-text
Library: transformers
Tags: transformers, pytorch, onnx, safetensors, vision-encoder-decoder, image-to-text, image-captioning, dataset:Mozilla/flickr30k-transformed-captions, base_model:google/vit-base-patch16-224-in21k, base_model:quantized:google/vit-base-patch16-224-in21k, license:apache-2.0, model-index, endpoints_compatible, region:us
Base models (quantized): id: google/vit-base-patch16-224-in21k
Card metadata: tags: image-to-text, image-captioning
license: apache-2.0
metrics: rouge
datasets: Mozilla/flickr30k-transformed-captions
widget: src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/savanna.jpg
example_title: Savanna, src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/football-match.jpg
example_title: Football Match, src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/airport.jpg
example_title: Airport
base_model: google/vit-base-patch16-224-in21k
model-index: name: mozilla/distilvit
results: task: type: image-to-text
name: Image To Text
dataset: name: Mozilla/flickr30k-transformed-captions
type: Mozilla/flickr30k-transformed-captions
metrics: name: ROUGE-1
type: rouge
value: 43.006
verified: no, name: ROUGE-2
type: rouge
value: 16.9939
verified: no, name: ROUGE-L
type: rouge
value: 38.8923
verified: no, name: ROUGE-LSUM
type: rouge
value: 38.8877
verified: no, name: loss
type: loss
value: 0.19939416646957397
verified: no, name: gen_len
type: gen_len
value: 11.327256736227712
verified: no
Likes: 0 | All-time downloads: 484
Created: 2024-06-19 13:20:21