A Novel Approach for Vehicle Identification based on Image Registration and Deep Learning

Document Type : Article

Authors

1 Faculty of Electrical Engineering, Shahrood University of Technology, Shahrood, Iran

2 School of Electrical Engineering, Iran University of Science and Technology, Tehran, Iran.

3 Department of Computer Engineering, Ferdowsi University of Mashhad, Mashhad, Iran

Abstract

Fine-grained vehicle type recognition using on-road cameras is among interesting topics in machine vision. It has several challenges like inter-class similarity, different viewing angles, and different lighting and weather conditions. This paper presents a novel approach for vehicle classification based on a novel augmentation method and deep learning. In the proposed smart augmentation, the vehicle images of each class are registered on the reference vehicles of all other classes and then added to the training set of that class. In this way, we will have a lot of new images which are very similar to both reference and target classes. This helps the CNN model to handle inter-class similarities very well. In the test phase, the input image is registered on every reference image in parallel and applied to the model. Finally, the winner is determined by summing up the provided scores of all models. The targeted data augmentation along with the proposed classification strategy has high recognition power and is capable of providing high accuracy using small CNNs or any other classification method without the need for large datasets. The proposed method achieved a recognition rate of 99.8% with only 150K parameters.

Keywords


  1. References

    1. Biglari, M., Soleimani, A., and Hassanpour, H. "A Cascaded Part-Based System for Fine-Grained Vehicle Classification". IEEE Transactions on Intelligent Transportation Systems. 19(1), pp. 273-283 (2018).
    2. Ma, Z., Chang, D., Xie, J., et al. "Fine-Grained Vehicle Classification With Channel Max Pooling Modified CNNs". IEEE Transactions on Vehicular Technology. 68(4), pp. 3224-3233 (2019).
    3. Wang, J., Zheng, H., Huang, Y., et al. "Vehicle Type Recognition in Surveillance Images From Labeled Web-Nature Data Using Deep Transfer Learning". IEEE Transactions on Intelligent Transportation Systems. 19(9), pp. 2913-2922 (2018).
    4. Huang, Y., Wu, R., Sun, Y., et al. "Vehicle Logo Recognition System Based on Convolutional Neural Networks With a Pretraining Strategy". IEEE Transactions on Intelligent Transportation Systems. 16(4), pp. 1951-1960 (2015).
    5. Psyllos, A.P., Anagnostopoulos, C.-N.E., and Kayafas, E. "Vehicle Logo Recognition Using a SIFT-Based Enhanced Matching Scheme". IEEE Transactions on Intelligent Transportation Systems. 11(2), pp. 322-328 (2010).
    6. Zhang, J., Xiao, W., Coifman, B., et al. "Vehicle Tracking and Speed Estimation From Roadside Lidar". IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing. 13, pp. 5597-5608 (2020).
    7. Nazemi, A., Azimifar, Z., Shafiee, M.J., et al. "Real-Time Vehicle Make and Model Recognition Using Unsupervised Feature Learning". IEEE Transactions on Intelligent Transportation Systems. 21(7), pp. 3080-3090 (2020).
    8. Cai, D., Chen, K., Qian, Y., et al. "Convolutional low-resolution fine-grained classification". Pattern Recognition Letters. 119, pp. 166-171 (2019).
    9. Ni, X. and Huttunen, H. "Vehicle Attribute Recognition by Appearance: Computer Vision Methods for Vehicle Type, Make and Model Classification". Journal of Signal Processing Systems. 93(4), pp. 357-368 (2020).
    10. Soon, F.C., Khaw, H.Y., Chuah, J.H., et al. "PCANet-Based Convolutional Neural Network Architecture for a Vehicle Model Recognition System". IEEE Transactions on Intelligent Transportation Systems. 20(2), pp. 749-759 (2019).
    11. Ren, S., He, K., Girshick, R., et al. "Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks". IEEE Trans Pattern Anal Mach Intell. 39(6), pp. 1137-1149 (2017).
    12. Yang, L., Luo, P., Change Loy, C., et al. "A large-scale car dataset for fine-grained categorization and verification". Proceedings of the IEEE conference on computer vision and pattern recognition. pp. 3973-3981 (2015).
    13. Gholamalinejad, H. and Khosravi, H. "IRVD: A Large-Scale Dataset for Classification of Iranian Vehicles in Urban Streets". Journal of AI and Data Mining. 9(1), pp. 1-9 (2021).
    14. Yu, S., Wu, Y., Li, W., et al. "A model for fine-grained vehicle classification based on deep learning". Neurocomputing. 257, pp. 97-103 (2017).
    15. Sochor, J., Herout, A., and Havel, J. "BoxCars: 3D Boxes as CNN Input for Improved Fine-Grained Vehicle Recognition". 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). pp. 3006-3015 (2016).
    16. Sochor, J., Spanhel, J., and Herout, A. "BoxCars: Improving Fine-Grained Recognition of Vehicles Using 3-D Bounding Boxes in Traffic Surveillance". IEEE Transactions on Intelligent Transportation Systems. 20(1), pp. 97-108 (2019).
    17. Huang, Y., Liang, B., Xie, W., et al. "Dual Domain Multi-Task Model for Vehicle Re-Identification". IEEE Transactions on Intelligent Transportation Systems. pp. 1-9 (2020).
    18. Simonyan, K. and Zisserman, A. "Very deep convolutional networks for large-scale image recognition". arXiv preprint arXiv:1409.1556. (2014).
    19. Liu, R., Yuan, Z., and Liu, T. "Learning TBox With a Cascaded Anchor-Free Network for Vehicle Detection". IEEE Transactions on Intelligent Transportation Systems. pp. 1-12 (2020).
    20. Meng, D., Li, L., Liu, X., et al. "Parsing-based view-aware embedding network for vehicle re-identification". Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 7103-7112 (2020).
    21. Liu, H., Tian, Y., Yang, Y., et al. "Deep relative distance learning: Tell the difference between similar vehicles". Proceedings of the IEEE conference on computer vision and pattern recognition. pp. 2167-2175 (2016).
    22. Zhu, X., Luo, Z., Fu, P., et al. "VOC-ReID: Vehicle re-identification based on vehicle-orientation-camera". Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops. pp. 602-603 (2020).
    23. Hsieh, J.-W., Chen, L.-C., and Chen, D.-Y. "Symmetrical SURF and Its Applications to Vehicle Detection and Vehicle Make and Model Recognition". IEEE Transactions on Intelligent Transportation Systems. 15(1), pp. 6-20 (2014).
    24. Biglari, M., Soleimani, A., and Hassanpour, H. "Partā€based recognition of vehicle make and model". IET Image Processing. 11(7), pp. 483-491 (2017).
    25. Madden, M. and Munroe, D.T. "Multi-Class and Single-Class Classification Approaches to Vehicle Model Recognition from Images". (2005).
    26. Boonsim, N. and Prakoonwit, S. "Car make and model recognition under limited lighting conditions at night". Pattern Analysis and Applications. 20(4), pp. 1195-1207 (2016).
    27. Sarfraz, M.S. and Khan, M.H. "A Probabilistic Framework for Patch based Vehicle Type Recognition". Visapp. 1(2011).
    28. Asgarian Dehkordi, R. and Khosravi, H. "Vehicle Type Recognition based on Dimension Estimation and Bag of Word Classification". Journal of AI and Data Mining. 8(3), pp. 427-438 (2020).
    29. Howard, A.G., Zhu, M., Chen, B., et al. "Mobilenets: Efficient convolutional neural networks for mobile vision applications". arXiv preprint arXiv:1704.04861. (2017).
    30. Tan, M. and Le, Q. "Efficientnet: Rethinking model scaling for convolutional neural networks". International Conference on Machine Learning. PMLR. pp. 6105-6114 (2019).
    31. Nagy, B., Foehn, P., and Scaramuzza, D. "Faster than FAST: GPU-Accelerated Frontend for High-Speed VIO". arXiv preprint arXiv:2003.13493. (2020).