Real-Time Classification of Vehicles Using Machine Learning Algorithm on the Extensive Dataset

Abstract

Vehicle classification (VC) is a prominent research domain within image processing and machine learning (ML) for identifying vehicle volumes and traffic rule violations. In developed countries, nearly 40% of daily accidents are fatal, while in developing countries, the figure rises to 70%. Traditionally, vehicle detection and classification have been performed manually by experts, which is difficult, time-consuming, and prone to errors. Furthermore, incorrect detection and classification can result in hazardous situations. This highlights the need for more reliable techniques to identify and classify vehicles accurately and practically. In existing applications, numerous automated methods have been proposed. However, employing deep and machine learning algorithms on complex datasets of vehicle images has failed to achieve accuracy in various climate conditions and has been time-consuming. This paper presents an accurate, robust, real-time system to classify vehicles from onsite roads. The proposed system utilizes a random wavelet transform for pre-processing, edge and region-based segmentation for feature extraction, an embedded method for feature selection, and the XGBoost algorithm for VC. The proposed work classifies vehicles under complex weather, illumination, color, and occlusion conditions over 10 datasets, including a novel dataset named SRM2KTR, containing 75,436 vehicle images on an FPGA platform. The results show 98.81% accuracy, outperforming the state-of-the-art (98%). The system was demonstrated with four different classifiers, classifying images in 0.16 ns with an average accuracy of 97.79%. The system exhibits high accuracy, rapid identification time, and robustness in practical use.

Keywords

• Vehicle classification
• machine learning
• eXtreme gradient boost algorithm