Citation

Tan, Shao Xian and Ong, Jia You and Goh, Michael Kah Ong and Tee, Connie (2024) Boosting Vehicle Classification with Augmentation Techniques across Multiple YOLO Versions. JOIV : International Journal on Informatics Visualization, 8 (1). p. 45. ISSN 2549-9610

Text 2313-6449-1-PB.pdf - Published Version Restricted to Repository staff only Download (4MB)

Abstract

In recent years, computer vision has experienced a surge in applications across various domains, including product and quality inspection, automatic surveillance, and robotics. This study proposes techniques to enhance vehicle object detection and classification using augmentation methods based on the YOLO (You Only Look Once) network. The primary objective of the trained model is to generate a local vehicle detection system for Malaysia which have the capacity to detect vehicles manufactured in Malaysia, adapt to the specific environmental factors in Malaysia, and accommodate varying lighting conditions prevalent in Malaysia. The dataset used for this paper to develop and evaluate the proposed system was provided by a highway company, which captured a comprehensive top-down view of the highway using a surveillance camera. Rigorous manual annotation was employed to ensure accurate annotations within the dataset. Various image augmentation techniques were also applied to enhance the dataset's diversity and improve the system's robustness. Experiments were conducted using different versions of the YOLO network, such as YOLOv5, YOLOv6, YOLOv7, and YOLOv8, each with varying hyperparameter settings. These experiments aimed to identify the optimal configuration for the given dataset. The experimental results demonstrated the superiority of YOLOv8 over other YOLO versions, achieving an impressive mean average precision of 97.9% for vehicle detection. Moreover, data augmentation effectively solves the issues of overfitting and data imbalance while providing diverse perspectives in the dataset. Future research can focus on optimizing computational efficiency for real-time applications and large-scale deployments.

Item Type:	Article
Uncontrolled Keywords:	Vehicle detection; vehicle classification; object detection; YOLO; computer vision
Subjects:	T Technology > TL Motor vehicles. Aeronautics. Astronautics > TL1-484 Motor vehicles. Cycles
Divisions:	Faculty of Information Science and Technology (FIST)
Depositing User:	Ms Nurul Iqtiani Ahmad
Date Deposited:	03 May 2024 02:22
Last Modified:	03 May 2024 02:22
URII:	http://shdl.mmu.edu.my/id/eprint/12418

Downloads

Edit (login required)