首页|期刊导航|计算机与现代化|双域特征融合和重校准小目标检测网络

双域特征融合和重校准小目标检测网络OA

Dual-domain Feature Fusion and Recalibration of Small Target Detection Network

中文摘要英文摘要

针对目标检测领域中小目标检测区域小、特征像素少、识别效果差等问题,提出一种基于RT-DETR改进的小目标检测算法DFFR-DETR.首先,提出空间频率双域特征融合模块来改进主干网络,命名为SDFF-Net,替换原主干网络,提取图像中小目标的结构特征,增强对小目标信息的捕捉能力.其次,引入重校准注意单元和RepConv设计边界聚合重参数化(BAR)模块,提升网络对多尺度特征的融合与提取能力.最后,基于BAR模块设计基于卷积神经网络的多尺度特征重校准模型,增强全局特征的提取能力,进一步改善小目标检测性能.在VisDrone2019数据集上的实验结果表明,DFFR-DETR模型在验证集和测试集上的mAP50分别达到了52.6%和41.3%,比基线模型提高了5.1百分点和4.0百分点,此外精确率、召回率也有不同程度的提升;在TinyPerson数据集上做了泛化性实验,DFFR-DETR模型相较于基线模型在召回率、mAP50上分别提升了3.9百分点和2.3百分点,验证了改进模型的有效性和泛化性.

To address the challenges in small target detection,such as limited detection regions,sparse feature pixels,and poor recognition performance,this paper proposes an improved small target detection algorithm DFFR-DETR based on RT-DETR.Firstly,a spatial-frequency dual-domain feature fusion module is proposed to improve the backbone network,named SDFF-Net,which replaces the original backbone network to extract the structural features of small targets in the image and enhance the ability to capture information about small targets.Secondly,the introduction of the recalibration attention unit and the RepConv design of the Boundary Aggregation Reparameterization(BAR)module enhances the network's ability to fuse and extract multi-scale features.Finally,based on the BAR module,a multi-scale feature recalibration model based on convolutional neural net-works is designed to enhance the extraction ability of global features and further improve the performance of small target detec-tion.The experimental results on the VisDrone2019 dataset show that the DFFR-DETR model achieves mAP50 scores of 52.6% and 41.3% on the validation set and test set respectively,which are 5.1 percentage points and 4.0 percentage points higher than the baseline model.Additionally,precision and recall rates also demonstrate notable enhancements.The generalization experi-ments are conducted on the TinyPerson dataset.Compared to the baseline model,the DFFR-DETR model achieved an increase of 3.9 percentage points in recall rate and 2.3 percentage points in mAP50,verifying the effectiveness and generalization ability of the improved model.

卢宇东;李华;任德均;程科然;张智勇

四川大学机械工程学院,四川 成都 610065四川大学机械工程学院,四川 成都 610065四川大学机械工程学院,四川 成都 610065四川大学机械工程学院,四川 成都 610065四川大学机械工程学院,四川 成都 610065

信息技术与安全科学

小目标检测RT-DETR特征融合特征金字塔

small target detectionRT-DETRfeature fusionfeature pyramid

《计算机与现代化》 2026 (2)

39-45,52,8

10.3969/j.issn.1006-2475.2026.02.005

评论