dima806 commited on
Commit
5cc0daf
·
verified ·
1 Parent(s): b97dac8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -20
README.md CHANGED
@@ -8,34 +8,34 @@ base_model:
8
  pipeline_tag: image-classification
9
  library_name: transformers
10
  ---
11
- Returns the clothes category based on an image.
12
 
13
  See https://www.kaggle.com/code/dima806/clothes-image-detection-vit for details.
14
 
15
- ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/HEQBcYrJblxjKj5Y1bo5q.png)
16
 
17
  ```
18
  Classification report:
19
 
20
  precision recall f1-score support
21
 
22
- Blazer 0.1398 0.1650 0.1514 200
23
- Coat 0.0468 0.0400 0.0431 200
24
- Denim Jacket 0.4834 0.3650 0.4160 200
25
- Dresses 0.0296 0.0300 0.0298 200
26
- Hoodie 0.0035 0.0050 0.0041 200
27
- Jacket 0.2879 0.1900 0.2289 200
28
- Jeans 0.3394 0.3700 0.3541 200
29
- Long Pants 0.2754 0.0950 0.1413 200
30
- Polo 0.3125 0.1500 0.2027 200
31
- Shirt 0.2094 0.2900 0.2432 200
32
- Shorts 0.1638 0.0950 0.1203 200
33
- Skirt 0.2105 0.0200 0.0365 200
34
- Sports Jacket 0.0394 0.0400 0.0397 200
35
- Sweater 0.1328 0.3300 0.1894 200
36
- T-shirt 0.1641 0.2650 0.2027 200
37
 
38
- accuracy 0.1633 3000
39
- macro avg 0.1892 0.1633 0.1602 3000
40
- weighted avg 0.1892 0.1633 0.1602 3000
41
  ```
 
8
  pipeline_tag: image-classification
9
  library_name: transformers
10
  ---
11
+ Returns the clothes category with about 87% accuracy based on an image.
12
 
13
  See https://www.kaggle.com/code/dima806/clothes-image-detection-vit for details.
14
 
15
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/nHatsaQxttuXETO_3XM8I.png)
16
 
17
  ```
18
  Classification report:
19
 
20
  precision recall f1-score support
21
 
22
+ Blazer 0.7419 0.6900 0.7150 200
23
+ Coat 0.7512 0.7550 0.7531 200
24
+ Denim Jacket 0.8592 0.9150 0.8862 200
25
+ Dresses 0.8603 0.7700 0.8127 200
26
+ Hoodie 0.6985 0.9500 0.8051 200
27
+ Jacket 0.7686 0.4650 0.5794 200
28
+ Jeans 0.8657 0.8700 0.8678 200
29
+ Long Pants 0.8112 0.7950 0.8030 200
30
+ Polo 0.7929 0.5550 0.6529 200
31
+ Shirt 0.7430 0.7950 0.7681 200
32
+ Shorts 0.9149 0.8600 0.8866 200
33
+ Skirt 0.8102 0.8750 0.8413 200
34
+ Sports Jacket 0.6562 0.7350 0.6934 200
35
+ Sweater 0.7758 0.8650 0.8180 200
36
+ T-shirt 0.7743 0.8750 0.8216 200
37
 
38
+ accuracy 0.7847 3000
39
+ macro avg 0.7883 0.7847 0.7803 3000
40
+ weighted avg 0.7883 0.7847 0.7803 3000
41
  ```