Back to Search Start Over

Bi-YOLO:An improved lightweight object detection algorithm based on YOLOv8n.

Authors :
LIU Zi-yang
XU Hui-ying
ZHU Xin-zhong
LI Chen
WANG Ze-yu
CAO Yu-qi
DAI Kang-ji
Source :
Computer Engineering & Science / Jisuanji Gongcheng yu Kexue; Aug2024, Vol. 46 Issue 8, p1444-1454, 11p
Publication Year :
2024

Abstract

The single-stage object detection technology represented by YOLOv8 has significant optimizations in the backbone network, but fails to efficiently integrate contextual information in the neck network, leading to missed and false detections in small object detection. Additionally, the large number of algorithm parameters and high computational complexity make it unsuitable for end-to-end industrial deployment. To address these issues, this paper introduce the BiFormer attention mechanism based on the Transformer structure to enhance the detection performance for small objects and improve the algorithm's accuracy. At the same time introduce the GSConv module to reduce the algorithm size while ensuring no adverse impact on its performance, balancing the increase in computational and parametric costs brought by BiFormer. An object detection algorithm named Bi-YOLO is designed to achieve a balance between lightweight and algorithm performance. Experimental results show that compared to YOLOv8n, the Bi-YOLO object detection algorithm improves algorithm accuracy by 4.6%, increases the small object detection accuracy on the DOTA dataset by 2.3%, and reduces the number of parameters by 12.5%. Bi-YOLO effectively achieves a balance between algorithm lightweight and performance, providing a new approach for end-to-end industrial deployment. [ABSTRACT FROM AUTHOR]

Details

Language :
Chinese
ISSN :
1007130X
Volume :
46
Issue :
8
Database :
Complementary Index
Journal :
Computer Engineering & Science / Jisuanji Gongcheng yu Kexue
Publication Type :
Academic Journal
Accession number :
179575158
Full Text :
https://doi.org/10.3969/j.issn.1007-130X.2024.08.013