dima806 commited on
Commit
a44f3ec
1 Parent(s): b2dfea9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -2
README.md CHANGED
@@ -9,6 +9,21 @@ base_model:
9
  - google/vit-base-patch16-224-in21k
10
  pipeline_tag: image-classification
11
  ---
12
- Detects gender based on an image.
13
 
14
- See https://www.kaggle.com/code/dima806/gender-image-classification-vit for details.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  - google/vit-base-patch16-224-in21k
10
  pipeline_tag: image-classification
11
  ---
12
+ Detects gender with about 88% accuracy based on an image.
13
 
14
+ See https://www.kaggle.com/code/dima806/gender-image-classification-vit for details.
15
+
16
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/NLKajM7gBl16afwtMGLpD.png)
17
+
18
+ ```
19
+ Classification report:
20
+
21
+ precision recall f1-score support
22
+
23
+ Female 0.8563 0.8938 0.8746 480
24
+ Male 0.8978 0.8615 0.8793 520
25
+
26
+ accuracy 0.8770 1000
27
+ macro avg 0.8770 0.8776 0.8770 1000
28
+ weighted avg 0.8779 0.8770 0.8770 1000
29
+ ```