IRVD: A Large-Scale Dataset for Classification of Iranian Vehicles in Urban Streets

Gholamalinejad, H.; Khosravi, H.

doi:10.22044/jadm.2020.8438.1982

Document Type : Original/Review Paper

Authors

H. Gholamalinejad ¹
H. Khosravi ²

¹ Faculty of Electrical Engineering and Robotics, Shahrood University of Technology, Shahrood, Iran.

² Faculty of Electrical Engineering and Robotics, Shahrood University of Technology, Shahrood, Iran

https://doi.org/10.22044/jadm.2020.8438.1982

Abstract

In recent years, vehicle classification has been one of the most important research topics. However, due to the lack of a proper dataset, this field has not been well developed as other fields of intelligent traffic management. Therefore, the preparation of large-scale datasets of vehicles for each country is of great interest. In this paper, we introduce a new standard dataset of popular Iranian vehicles. This dataset, which consists of images from moving vehicles in urban streets and highways, can be used for vehicle classification and license plate recognition. It contains a large collection of vehicle images in different dimensions, viewing angles, weather, and lighting conditions. It took more than a year to construct this dataset. Images are taken from various types of mounted cameras, with different resolutions and at different altitudes. To estimate the complexity of the dataset, some classic methods alongside popular Deep Neural Networks are trained and evaluated on the dataset. Furthermore, two light-weight CNN structures are also proposed. One with 3-Conv layers and another with 5-Conv layers. The 5-Conv model with 152K parameters reached the recognition rate of 99.09% and can process 48 frames per second on CPU which is suitable for real-time applications.

Keywords

20.1001.1.23225211.2021.9.1.1.2

References

[1] H. O. S. D. Branch, “Imagery library for intelligent detection systems (i-lids),” in 2006 IET Conference on Crime and Security, 2006: IET, pp. 445-448.

[2] Y. Peng, J. S. Jin, S. Luo, M. Xu, and Y. Cui, “Vehicle type classification using PCA with self-clustering, ” in Multimedia and Expo Workshops (ICMEW), 2012 IEEE International Conference on, 2012: IEEE, pp. 384-389.

[3] Z. Dong, M. Pei, Y. He, T. Liu, Y. Dong, and Y. Jia, “Vehicle Type Classification Using Unsupervised Convolutional Neural Network, ” presented at the Proceedings of the 2014 22nd International Conference on Pattern Recognition, 2014.

[4] H. S. Bharadwaj, S. Biswas, and K. Ramakrishnan, “A large scale dataset for classification of vehicles in urban traffic scenes, ” in Proceedings of the Tenth Indian Conference on Computer Vision, Graphics and Image Processing, Indian 2016: ACM, p. 83.

[5] M. Biglari, A. Soleimani, and H. Hassanpour, “Using discriminative part for vehicle make and model recognition, ” Signal and data processing, vol. 15, pp. 41-54, 2018.

[6] J. Sochor, J. Špaňhel, and A. Herout, “Boxcars: Improving fine-grained recognition of vehicles using 3-d bounding boxes in traffic surveillance, ” IEEE transactions on intelligent transportation systems, vol. 20, no. 1, pp. 97-108, 2018.

[7] R. Asgarian Dehkordi and H. Khosravi, “Vehicle Type Recognition based on Dimension Estimation and Bag of Word Classification, ” Journal of AI and Data Mining, pp. -, 2020, doi: 10.22044/jadm.2020.8375.1975.

[8] “SATPA license plate recognition library. ” https://shahaab-co.ir/license-plate-recognition-library/ (accessed 2020/04/20.

[9] J. Redmon, S. Divvala, R. Girshick, and A. Farhadi, “You only look once: Unified, real-time object detection, ” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 779-788.

[10] N. Dalal and B. Triggs, “Histograms of oriented gradients for human detection, ” in international Conference on computer vision & Pattern Recognition (CVPR'05), 2005, vol. 1: IEEE Computer Society, pp. 886--893.

[11] R. Dechter, Learning while searching in constraint-satisfaction problems. University of California, Computer Science Department, Cognitive Systems …, 1986.

[12] K. Simonyan and A. Zisserman, “Very deep convolutional networks for large-scale image recognition, ” arXiv preprint arXiv:1409.1556, 2014.

[13] C. Francois, “Deep learning with Python," ed: Manning Publications Company, 2017.

[14] K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition, ” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2016, pp. 770-778.

[15] J. Redmon. “Darknet: Open Source Neural Networks. ”https://github.com/pjreddie/darknet (accessed 2020/04/24.

[16] J. Redmon and A. Farhadi, “Yolov3: An incremental improvement, ” arXiv preprint arXiv:1804.02767, 2018.

[17] F. Branchaud-Charron, A. Achkar, and P.-M. Jodoin, “Spectral metric for dataset complexity assessment, ” in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2019, pp. 3215-3224.

IRVD: A Large-Scale Dataset for Classification of Iranian Vehicles in Urban Streets

References

References

Volume 9, Issue 1January 2021Pages 1-9

Volume 9, Issue 1
January 2021
Pages 1-9