Open Access iconOpen Access

ARTICLE

crossmark

A Particle Swarm Optimization Based Deep Learning Model for Vehicle Classification

Adi Alhudhaif1,*, Ammar Saeed2, Talha Imran2, Muhammad Kamran3, Ahmed S. Alghamdi3, Ahmed O. Aseeri1, Shtwai Alsubai1

1 Department of Computer Science, College of Computer Engineering and Sciences in Al-Kharj, Prince Sattam bin Abdulaziz University, P.O. Box 151, Al-Kharj, 11942, Saudi Arabia
2 Department of Computer Science, COMSATS University Islamabad, Wah Cantt, 47010, Pakistan
3 University of Jeddah, College of Computer Science and Engineering Department of Cybersecurity, Jeddah, 21959, Saudi Arabia

* Corresponding Author: Adi Alhudhaif. Email: email

Computer Systems Science and Engineering 2022, 40(1), 223-235. https://doi.org/10.32604/csse.2022.018430

Abstract

Image classification is a core field in the research area of image processing and computer vision in which vehicle classification is a critical domain. The purpose of vehicle categorization is to formulate a compact system to assist in real-world problems and applications such as security, traffic analysis, and self-driving and autonomous vehicles. The recent revolution in the field of machine learning and artificial intelligence has provided an immense amount of support for image processing related problems and has overtaken the conventional, and handcrafted means of solving image analysis problems. In this paper, a combination of pre-trained CNN GoogleNet and a nature-inspired problem optimization scheme, particle swarm optimization (PSO), was employed for autonomous vehicle classification. The model was trained on a vehicle image dataset obtained from Kaggle that has been suitably augmented. The trained model was classified using several classifiers; however, the Cubic SVM (CSVM) classifier was found to outperform the others in both time consumption and accuracy (94.8%). The results obtained from empirical evaluations and statistical tests reveal that the model itself has shown to outperform the other related models not only in terms of accuracy (94.8%) but also in terms of training time (82.7 s) and speed prediction (380 obs/sec).

Keywords


Cite This Article

A. Alhudhaif, A. Saeed, T. Imran, M. Kamran, A. S. Alghamdi et al., "A particle swarm optimization based deep learning model for vehicle classification," Computer Systems Science and Engineering, vol. 40, no.1, pp. 223–235, 2022.

Citations




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.
  • 3209

    View

  • 1860

    Download

  • 1

    Like

Share Link