metadata
license: apache-2.0
metrics:
- accuracy
- f1
base_model:
- google/vit-base-patch16-224-in21k
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