IEEE Access (Jan 2020)

Vehicle Type Classification Using an Enhanced Sparse-Filtered Convolutional Neural Network With Layer-Skipping Strategy

  • Suryanti Awang,
  • Nik Mohamad Aizuddin Nik Azmi,
  • Md. Arafatur Rahman

DOI
https://doi.org/10.1109/ACCESS.2019.2963486
Journal volume & issue
Vol. 8
pp. 14265 – 14277

Abstract

Read online

In this paper, a vehicle type classification approach is proposed by using an enhanced feature extraction technique based on Sparse-Filtered Convolutional Neural Network with Layer-Skipping strategy (SF-CNNLS). To extract rich and discriminant vehicle features, we introduce Three-Channels of SF-CNNLS (TC-SF-CNNLS) as the feature extraction technique. Local and global features of vehicles are extracted from three channels of an image which are, luminance and chromatic components. This technique is inspired by how human eyes differentiating objects that share almost similar features. TC-SF-CNNLS is tested with a benchmark dataset that provides frontal-view images to classify vehicle types of the bus, passenger car, taxi, minivan, SUV, and truck with Softmax Regression as a classifier. This test aims to observe the ability of this technique in differentiating vehicles with almost similar features but different classes. A test is also conducted with the self-obtained dataset (SPINT) to observe the effectiveness of this technique. The results are observed based on accuracy, precision, recall, and f-score, whereby, TCSF-NNLS has successfully recognized all the classes with an average accuracy of 0.905, precision is between 0.8629 to 0.9548, recall is between 0.83 to 0.96 and f-score is between 0.8564 to 0.9523. In addition, this technique is able to outperform other existing techniques with an average accuracy of 93.% compared to only 89.2% when 5 classes of vehicles are tested.

Keywords