Lightweight object detection algorithm for road scenes based on YOLOV8

Chengqin Huang; Degang Yang; Xin Zhang

doi:10.1117/12.3029017

22 May 2024 Lightweight object detection algorithm for road scenes based on YOLOV8

Chengqin Huang, Degang Yang, Xin Zhang

Proceedings Volume 13176, Fourth International Conference on Machine Learning and Computer Application (ICMLCA 2023); 131762M (2024) https://doi.org/10.1117/12.3029017
Event: Fourth International Conference on Machine Learning and Computer Application (ICMLCA 2023), 2023, Hangzhou, China

Abstract

Object detection in road scenes is a crucial component of autonomous driving. Due to the significant variations in target scales, it is prone to false positives, false negatives, and some underperforming devices cannot deploy the state-of-the-art detectors. To address these issues, we propose a lightweight algorithm based on an improved YOLOv8. We simplify the model by using the FasterNet Block from FasterNet to replace the BottleNeck module in YOLOv8's backbone network C2f, reducing parameters and floating-point computations. We also enhance feature extraction by substituting RFCAConv for the downsampling standard convolution in C2f. Additionally, we introduce Wise-IoU to replace the original activation function, directing the network's focus towards anchor boxes of average quality.To promote effective fusion of original, shallow, and deep features, we introduce the BiFPN structure to replace YOLOv8's PAN structure. Furthermore, a small object detection layer is added to the head to handle the drastic scale variations in road scenes. Experimental results on the SODA10M dataset demonstrate that the improved YOLOv8 model achieves a 55.8% mAP@0.5 and a 35.5% mAP@0.5:0.95. The model's parameter count, size, and floating-point computations decrease by 58.3%, 57.0%, and 21.1%, respectively. Analysis of the experimental results confirms that the proposed model is effective and superior, striking a balance between detection accuracy and model lightweightness.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Chengqin Huang, Degang Yang, and Xin Zhang "Lightweight object detection algorithm for road scenes based on YOLOV8", Proc. SPIE 13176, Fourth International Conference on Machine Learning and Computer Application (ICMLCA 2023), 131762M (22 May 2024); https://doi.org/10.1117/12.3029017

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Roads

Detection and tracking algorithms

Data modeling

Feature fusion

Convolution

Head

Show All Keywords

Keywords/Phrases

Search In:

Publication Years