Hugging Face's logo

dima806
/
shoe_types_image_detection

Image Classification
Transformers
PyTorch
Safetensors
vit
Model card Files
xet
Community
shoe_types_image_detection
File size: 863 Bytes 
0e72d54
 
 
 
 
10a9a52
 
0e72d54
036ef78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
---
license: apache-2.0
metrics:
- accuracy
- f1
base_model:
- google/vit-base-patch16-224-in21k
---
Return shoe type given an image.

![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/VRp8h5fKGtsdi_ydGM2Oj.png)

See https://www.kaggle.com/code/dima806/shoe-type-image-detection-vit for more details.

```
Classification report:

              precision    recall  f1-score   support

        Clog     0.9748    0.9598    0.9672      1169
      Brogue     0.9804    0.9812    0.9808      1170
     Sneaker     0.9718    0.9735    0.9727      1170
        Boat     0.9642    0.9658    0.9650      1170
 Ballet Flat     0.9729    0.9837    0.9783      1169

    accuracy                         0.9728      5848
   macro avg     0.9728    0.9728    0.9728      5848
weighted avg     0.9728    0.9728    0.9728      5848
```