A vehicle classification method based on deep learning and data augmentation

Kaiyan Zhang; Xinglong Feng

doi:10.1117/12.2678655

14 April 2023 A vehicle classification method based on deep learning and data augmentation

Kaiyan Zhang, Xinglong Feng

Proceedings Volume 12634, International Conference on Optics and Machine Vision (ICOMV 2023); 1263408 (2023) https://doi.org/10.1117/12.2678655
Event: International Conference on Optics and Machine Vision (ICOMV 2023), 2023, Changsha, China

Abstract

At present, deep learning has been applied in a lot of autonomous driving, and vehicle classification based on deep learning is an important content. Although deep learning methods have proven to be more effective at classifying vehicles than traditional machine learning methods. However, in practice, due to certain similar characteristics among different types of vehicles, the accuracy of vehicle classification based on deep learning method is not high enough. In order to improve the effectiveness of deep learning in the field of vehicle classification, this paper studies from the data side. The method of this paper is to propose a novel data augmentation method according to the characteristics of vehicles and combine with ResNet34 model. After experimental verification, the results of the test set show that the classification accuracy of the ResNet34 model after data augmentation in this paper is 80.0%, higher than the classification accuracy of 75.42% without data augmentation. The above results show that the data augmentation method proposed in this paper is very effective for vehicle classification problems and can be used in conjunction with deep learning model.

Citation Download Citation

Kaiyan Zhang and Xinglong Feng "A vehicle classification method based on deep learning and data augmentation", Proc. SPIE 12634, International Conference on Optics and Machine Vision (ICOMV 2023), 1263408 (14 April 2023); https://doi.org/10.1117/12.2678655

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available