Vehicle Detection and Classification via YOLOv8 and Deep Belief Network over Aerial Image Sequences

Naif Al Mudawi, Asifa Mehmood Qureshi, Maha Abdelhaq, Abdullah Alshahrani, Abdulwahab Alazeb, Mohammed Alonazi, Asaad Algarni

Research output: Contribution to journalArticlepeer-review

32 Scopus citations

Abstract

Vehicle detection and classification are the most significant and challenging activities of an intelligent traffic monitoring system. Traditional methods are highly computationally expensive and also impose restrictions when the mode of data collection changes. This research proposes a new approach for vehicle detection and classification over aerial image sequences. The proposed model consists of five stages. All of the images are preprocessed in the first stage to reduce noise and raise the brightness level. The foreground items are then extracted from these images using segmentation. The segmented images are then passed onto the YOLOv8 algorithm to detect and locate vehicles in each image. The feature extraction phase is then applied to the detected vehicles. The extracted feature involves Scale Invariant Feature Transform (SIFT), Oriented FAST and Rotated BRIEF (ORB), and KAZE features. For classification, we used the Deep Belief Network (DBN) classifier. Based on classification, the experimental results across the three datasets produced better outcomes; the proposed model attained an accuracy of 95.6% over Vehicle Detection in Aerial Imagery (VEDAI) and 94.6% over Vehicle Aerial Imagery from a Drone (VAID) dataset, respectively. To compare our model with the other standard techniques, we have also drawn a comparative analysis with the latest techniques in the research.

Original languageEnglish
Article number14597
JournalSustainability (Switzerland)
Volume15
Issue number19
DOIs
StatePublished - Oct 2023

Keywords

  • DBN
  • YOLOv5
  • classification
  • segmentation
  • vehicle detection

Fingerprint

Dive into the research topics of 'Vehicle Detection and Classification via YOLOv8 and Deep Belief Network over Aerial Image Sequences'. Together they form a unique fingerprint.

Cite this