Bi-YOLO:An improved lightweight object detection algorithm based on YOLOv8n
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.
国家科技期刊平台
NSTL
万方数据
