Update README.md
Browse files
README.md
CHANGED
@@ -4,4 +4,27 @@ metrics:
|
|
4 |
- accuracy
|
5 |
- f1
|
6 |
---
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
- accuracy
|
5 |
- f1
|
6 |
---
|
7 |
+
Return a vehicle type probability based on an image.
|
8 |
+
|
9 |
+
See https://www.kaggle.com/code/dima806/vehicle-10-types-detection-vit for more details.
|
10 |
+
|
11 |
+
```
|
12 |
+
Classification report:
|
13 |
+
|
14 |
+
precision recall f1-score support
|
15 |
+
|
16 |
+
SUV 0.8780 0.9000 0.8889 40
|
17 |
+
bus 1.0000 1.0000 1.0000 40
|
18 |
+
family sedan 0.8571 0.9000 0.8780 40
|
19 |
+
fire engine 0.8444 0.9500 0.8941 40
|
20 |
+
heavy truck 0.9459 0.8750 0.9091 40
|
21 |
+
jeep 0.9512 0.9750 0.9630 40
|
22 |
+
minibus 0.9500 0.9500 0.9500 40
|
23 |
+
racing car 1.0000 0.9500 0.9744 40
|
24 |
+
taxi 0.9750 0.9750 0.9750 40
|
25 |
+
truck 0.9722 0.8750 0.9211 40
|
26 |
+
|
27 |
+
accuracy 0.9350 400
|
28 |
+
macro avg 0.9374 0.9350 0.9354 400
|
29 |
+
weighted avg 0.9374 0.9350 0.9354 400
|
30 |
+
```
|