Downloads · 30 days
19
4% of all-time downloads
marvel_heroes_image_detection
dima806/marvel_heroes_image_detection
marvel_heroes_image_detection is a image classification model from dima806. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Return Marvel hero based on image with about 88% accuracy.
All-time downloads
533
Public
Parameters
85.8M
3.1 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt2.1 GB · 55%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Sep 19, 2023
- Updated
- Oct 19, 2024
- SHA
- a070e9cf
Base models
Model card
From the Hugging Face model README
Return Marvel hero based on image with about 88% accuracy.
See https://www.kaggle.com/code/dima806/marvel-heroes-image-detection-vit for more details.
Classification report:
precision recall f1-score support
captain america 0.8519 0.8519 0.8519 162
black widow 0.8634 0.8528 0.8580 163
spider-man 0.9571 0.9630 0.9600 162
thanos 0.8917 0.8589 0.8750 163
ironman 0.8614 0.8827 0.8720 162
hulk 0.8889 0.8395 0.8635 162
loki 0.8957 0.8957 0.8957 163
doctor strange 0.8629 0.9264 0.8935 163
accuracy 0.8838 1300
macro avg 0.8841 0.8838 0.8837 1300
weighted avg 0.8841 0.8838 0.8837 1300
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Sep 19, 2023
- Updated
- Oct 19, 2024