| license: apache-2.0 | |
| # Model for Predicting Car Orientation from Images | |
| ### Description | |
| - **Repository:** [Car Orientation Prediction Repository](https://github.com/fort-cyber/car-orientation) | |
| - **Paper:** [Car Orientation Prediction Paper](https://www.mdpi.com/2079-9292/12/24/4947) | |
| ### Model Description | |
| 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. | |
| It was pre-trained on ImageNet and fine-tuned on the CFV Dataset, consisting of 23K images of vehicles with diverse orientations. | |
| 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. | |
| ### Citation Information | |
| If you utilize this model for any project or research, please cite our paper: | |
| ``` | |
| @article{catruna2023car, | |
| title={Car Full View Dataset: Fine-Grained Predictions of Car Orientation from Images}, | |
| author={Catruna, Andy and Betiu, Pavel and Tertes, Emanuel and Ghita, Vladimir and Radoi, Emilian and Mocanu, Irina and Dascalu, Mihai}, | |
| journal={Electronics}, | |
| volume={12}, | |
| number={24}, | |
| pages={4947}, | |
| year={2023}, | |
| publisher={MDPI} | |
| } | |
| ``` |