| | --- |
| | 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 |
| | ``` |
