andycatruna commited on
Commit
a8fddad
·
1 Parent(s): 8121645

add readme and model weights

Browse files
Files changed (2) hide show
  1. README.md +28 -0
  2. 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