计算机工程与科学2024，Vol.46Issue(8)：1444-1454,11.DOI:10.3969/j.issn.1007-130X.2024.08.013
Bi-YOLO:-种基于YOLOv8n改进的轻量化目标检测算法
Bi-YOLO:An improved lightweight object detection algorithm based on YOLOv8n
摘要
Abstract
The single-stage object detection technology represented by YOLOv8 has significant opti-mizations 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 num-ber of algorithm parameters and high computational complexity make it unsuitable for end-to-end indus-trial 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 bal-ance 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 parame-ters by 12.5％.Bi-YOLO effectively achieves a balance between algorithm lightweight and perform-ance,providing a new approach for end-to-end industrial deployment.关键词
YOLOv8/BiFormer/轻量化改进/目标检测/端到端工业部署Key words
YOLOv8/BiFormer/lightweight improvement/object detection/end-to-end industrial de-ployment分类
信息技术与安全科学引用本文复制引用
刘子洋,徐慧英,朱信忠,李琛,王泽宇,曹雨淇,戴康佳..Bi-YOLO:-种基于YOLOv8n改进的轻量化目标检测算法[J].计算机工程与科学,2024,46(8):1444-1454,11.基金项目
国家自然科学基金(62376252,61976196) （62376252,61976196）
浙江省自然科学基金重点项目(LZ22F030003) （LZ22F030003）
国家级大学生创新训练计划重点项目(202310345042) （202310345042）
