Downloads · 30 days
39
0% of all-time downloads
headgear_image_detection
dima806/headgear_image_detection
headgear_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.
All-time downloads
11.1K
Public
Parameters
85.8M
2.1 GB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.pt1.4 GB · 50%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Oct 13, 2023
- Updated
- Oct 19, 2024
- SHA
- f13c493d
Base models
Model card
From the Hugging Face model README
Returns headgear type given an image.
See https://www.kaggle.com/code/dima806/headgear-image-detection-vit for more details.
Classification report:
precision recall f1-score support
BERET 1.0000 0.9565 0.9778 115
FEDORA 0.9913 1.0000 0.9956 114
SOMBERO 1.0000 1.0000 1.0000 115
HARD HAT 1.0000 1.0000 1.0000 115
FEZ 1.0000 0.9912 0.9956 114
ZUCCHETTO 1.0000 0.9912 0.9956 114
TOP HAT 1.0000 1.0000 1.0000 115
DEERSTALKER 0.9913 1.0000 0.9956 114
ASCOT CAP 0.9500 1.0000 0.9744 114
PORK PIE 0.9739 0.9825 0.9782 114
MILITARY HELMET 1.0000 1.0000 1.0000 115
BICORNE 1.0000 0.9912 0.9956 114
FOOTBALL HELMET 1.0000 1.0000 1.0000 115
MOTARBOARD 0.9913 1.0000 0.9956 114
BOATER 1.0000 1.0000 1.0000 115
PITH HELMET 0.9913 1.0000 0.9956 114
SOUTHWESTER 1.0000 0.9912 0.9956 114
BOWLER 0.9912 0.9825 0.9868 114
GARRISON CAP 1.0000 0.9912 0.9956 114
BASEBALL CAP 1.0000 1.0000 1.0000 115
accuracy 0.9939 2288
macro avg 0.9940 0.9939 0.9939 2288
weighted avg 0.9940 0.9939 0.9939 2288
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Oct 13, 2023
- Updated
- Oct 19, 2024