首页|期刊导航|计算机工程|BS-YOLO:基于BSAM注意力机制和SCConv的小目标检测算法

BS-YOLO:基于BSAM注意力机制和SCConv的小目标检测算法OA

BS-YOLO:A Small Object Detection Algorithm Based on BSAM Attention Mechanism and SCConv

中文摘要英文摘要

近年来,基于深度学习的目标检测算法在准确率和鲁棒性等方面取得了巨大进步,并且在工业界得到广泛应用.但是,在小目标检测领域,当前的目标检测算法仍然存在漏检率和误检率高的问题.因此,提出一种基于SCConv和BSAM注意力机制的YOLO小目标检测算法BS-YOLO.首先,针对特征提取网络存在大量冗余信息的问题,利用SCConv重构主干网络,提出一种新的模块C3SC,对提取到的特征图从空间和通道两个方面减少冗余信息,提升主干网络提取到的特征图质量,从而提高检测精度;其次,结合CBAM和BiFormer自注意力机制提出一种新的注意力机制BSAM,在空间和通道两个方面合理分配权重,使特征图更加关注有效信息,抑制背景的干扰;最后,为了解决小目标检测存在的难易样本分布不均的问题,利用Slideloss优化损失函数,从而提高小目标检测的效果.在RSOD数据集上的实验结果表明,BS-YOLO算法的精确率为94.2%,召回率为91.6%,均值平均精度(mAP@0.5)为95.9%,相对于原始的YOLOv5算法,分别提高了 3.3、0.1、3.6百分点,表明BS-YOLO算法可以有效提高小目标检测的精度,降低漏检率.

In recent years,there has been significant progress in terms of accuracy and robustness of deep-learning-based algorithms for object detection that have been widely applied in industry.However,in the field of small object detection,currently used object detection algorithms suffer from high rates of missed detections and false positives.Therefore,in this study,a YOLO small object detection algorithm,viz.,BS-YOLO,which is based on SCConv and BSAM attention mechanism,is developed.First,in response to the problem of the large amount of redundant information generated in the feature extraction network,a new module,viz.,C3SC,is proposed to reconstruct the backbone network using SCConv.This module reduces redundant information in both spatial and channel aspects of the extracted feature maps,thereby improving the quality of the feature maps extracted by the backbone network,and in turn enhancing detection accuracy.Second,a new attention mechanism,viz.,BSAM,is proposed by combining CBAM and the BiFormer self-attention mechanism,by which weights are allocated reasonably in both spatial and channel aspects,making the feature map more focused on effective information and suppressing background interference.Finally,to solve the problem of uneven distribution of difficult and easy samples in terms of small object detection,Slideloss is used to optimize the loss function,thereby improving the effectiveness of the algorithm for small object detection.The experimental results obtained using the RSOD dataset show that the BS-YOLO algorithm has a precision of 94.2%,a recall rate of 91.6%,and a mAP@0.5 of 95.9%,corresponding to improvements of 3.3,0.1,and 3.6 percentage,respectively,compared to the original YOLOv5 algorithm.This indicates that the BS-YOLO algorithm can effectively improve the accuracy of small object detection and reduce the missed detection rate.

曹继卫;罗飞;丁炜超

华东理工大学信息科学与工程学院,上海 200000||上海市计算机软件评测重点实验室,上海 200000华东理工大学信息科学与工程学院,上海 200000华东理工大学信息科学与工程学院,上海 200000

信息技术与安全科学

小目标检测注意力机制特征提纯计算机视觉深度学习

small object detectionattention mechanismfeature purificationcomputer visiondeep learning

《计算机工程》 2026 (3)

119-127,9

上海市自然科学基金(22ZR1416500).

10.19678/j.issn.1000-3428.0070159

评论