Open Access iconOpen Access

ARTICLE

crossmark

An Optimized Approach to Vehicle-Type Classification Using a Convolutional Neural Network

Shabana Habib1, Noreen Fayyaz Khan2,*

1 Department of Information Technology, College of Computer, Qassim University, Buraidah, 51452, Saudi Arabia
2 Department of Computer Science, Islamia College University, Peshawar, Pakistan

* Corresponding Author: Noreen Fayyaz Khan. Email: email

Computers, Materials & Continua 2021, 69(3), 3321-3335. https://doi.org/10.32604/cmc.2021.015504

Abstract

Vehicle type classification is considered a central part of an intelligent traffic system. In recent years, deep learning had a vital role in object detection in many computer vision tasks. To learn high-level deep features and semantics, deep learning offers powerful tools to address problems in traditional architectures of handcrafted feature-extraction techniques. Unlike other algorithms using handcrated visual features, convolutional neural network is able to automatically learn good features of vehicle type classification. This study develops an optimized automatic surveillance and auditing system to detect and classify vehicles of different categories. Transfer learning is used to quickly learn the features by recording a small number of training images from vehicle frontal view images. The proposed system employs extensive data-augmentation techniques for effective training while avoiding the problem of data shortage. In order to capture rich and discriminative information of vehicles, the convolutional neural network is fine-tuned for the classification of vehicle types using the augmented data. The network extracts the feature maps from the entire dataset and generates a label for each object (vehicle) in an image, which can help in vehicle-type detection and classification. Experimental results on a public dataset and our own dataset demonstrated that the proposed method is quite effective in detection and classification of different types of vehicles. The experimental results show that the proposed model achieves 96.04% accuracy on vehicle type classification.

Keywords


Cite This Article

APA Style
Habib, S., Khan, N.F. (2021). An optimized approach to vehicle-type classification using a convolutional neural network. Computers, Materials & Continua, 69(3), 3321-3335. https://doi.org/10.32604/cmc.2021.015504
Vancouver Style
Habib S, Khan NF. An optimized approach to vehicle-type classification using a convolutional neural network. Comput Mater Contin. 2021;69(3):3321-3335 https://doi.org/10.32604/cmc.2021.015504
IEEE Style
S. Habib and N.F. Khan, "An Optimized Approach to Vehicle-Type Classification Using a Convolutional Neural Network," Comput. Mater. Contin., vol. 69, no. 3, pp. 3321-3335. 2021. https://doi.org/10.32604/cmc.2021.015504



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1516

    View

  • 863

    Download

  • 0

    Like

Share Link