Update README.md
Browse files
README.md
CHANGED
@@ -4,4 +4,25 @@ metrics:
|
|
4 |
- accuracy
|
5 |
- f1
|
6 |
---
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
- accuracy
|
5 |
- f1
|
6 |
---
|
7 |
+
Return Marvel hero based on image.
|
8 |
+
|
9 |
+
See https://www.kaggle.com/code/dima806/marvel-heroes-image-detection-vit for more details.
|
10 |
+
|
11 |
+
```
|
12 |
+
Classification report:
|
13 |
+
|
14 |
+
precision recall f1-score support
|
15 |
+
|
16 |
+
captain america 0.8519 0.8519 0.8519 162
|
17 |
+
black widow 0.8634 0.8528 0.8580 163
|
18 |
+
spider-man 0.9571 0.9630 0.9600 162
|
19 |
+
thanos 0.8917 0.8589 0.8750 163
|
20 |
+
ironman 0.8614 0.8827 0.8720 162
|
21 |
+
hulk 0.8889 0.8395 0.8635 162
|
22 |
+
loki 0.8957 0.8957 0.8957 163
|
23 |
+
doctor strange 0.8629 0.9264 0.8935 163
|
24 |
+
|
25 |
+
accuracy 0.8838 1300
|
26 |
+
macro avg 0.8841 0.8838 0.8837 1300
|
27 |
+
weighted avg 0.8841 0.8838 0.8837 1300
|
28 |
+
```
|