andycatruna
commited on
Commit
·
a8fddad
1
Parent(s):
8121645
add readme and model weights
Browse files- README.md +28 -0
- convnext-multi-weak.pth +3 -0
README.md
CHANGED
@@ -1,3 +1,31 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
---
|
4 |
+
|
5 |
+
# Model for Predicting Car Orientation from Images
|
6 |
+
|
7 |
+
### Description
|
8 |
+
- **Repository:** [Car Orientation Prediction Repository](https://github.com/fort-cyber/car-orientation)
|
9 |
+
- **Paper:** [Car Orientation Prediction Paper](https://www.mdpi.com/2079-9292/12/24/4947)
|
10 |
+
|
11 |
+
### Model Description
|
12 |
+
This model is a ConvNext architecture fine-tuned for predicting car orientations from images. It represents the top-performing model in our experiments from the CFV Dataset paper.
|
13 |
+
|
14 |
+
It was pre-trained on ImageNet and fine-tuned on the CFV Dataset, consisting of 23K images of vehicles with diverse orientations.
|
15 |
+
|
16 |
+
The model expects a batch of images with sizes 224x224 and will output a vector consisting of scalar values that represent the pitch angle in degrees.
|
17 |
+
|
18 |
+
### Citation Information
|
19 |
+
If you utilize this model for any project or research, please cite our paper:
|
20 |
+
```
|
21 |
+
@article{catruna2023car,
|
22 |
+
title={Car Full View Dataset: Fine-Grained Predictions of Car Orientation from Images},
|
23 |
+
author={Catruna, Andy and Betiu, Pavel and Tertes, Emanuel and Ghita, Vladimir and Radoi, Emilian and Mocanu, Irina and Dascalu, Mihai},
|
24 |
+
journal={Electronics},
|
25 |
+
volume={12},
|
26 |
+
number={24},
|
27 |
+
pages={4947},
|
28 |
+
year={2023},
|
29 |
+
publisher={MDPI}
|
30 |
+
}
|
31 |
+
```
|
convnext-multi-weak.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:26bd9e4eff8a4cfa3a264223ecced760f2fb3d1f29dc68f4e23b1b27babcd91d
|
3 |
+
size 201141385
|