基于改进RT-DETR无人机红外小目标检测

UAV Infrared Small Target Detection Based on Improved RT-DET

  • 摘要: 针对无人机红外图像中小目标检测中因目标尺寸微小、信噪比低及复杂背景干扰导致的特征提取困难和检测精度不足问题,提出了一种改进的RT-DETR算法,记作MPH-DETR。首先,设计多尺度注意力特征增强模块,通过跨层特征融合与自适应注意力机制,增强对微弱目标的感知能力,同时减少模型参数。其次,在颈部网络中,提出Pola-SEFN结构提升模型的检测精度,解耦目标与背景信号,优化特征传递;最后,在特征融合环节设计Hyper-MFM结构,解决跨尺度语义割裂与特征丢失问题。实验结果表明,与RT-DETR基线相比,MPH-DETR在HIT-UAV红外数据集上,准确率、mAP@50和mAP@50:95分别提升了8.4%、5.8%和4.8%,有效改善复杂背景下的误检和漏检问题。

     

    Abstract: To address the problems of difficult feature extraction and insufficient detection accuracy when detecting small targets in infrared images obtained using unmanned aerial vehicles (UAVs) owing to small target sizes, low signal-to-noise ratios, and complex background interference, this study investigated an improved RT-DETR algorithm, denoted as MPH-DETR. First, a multiscale adaptive feature extraction module was designed, which simultaneously enhanced the ability to perceive weak targets and reduced the model parameters by means of cross-layer feature fusion and an adaptive attention mechanism. Second, in the neck network, a Pola-SEFN structure was proposed to improve the detection accuracy of the model, which decoupled the target and background signals and optimized the feature transmission. Finally, in the feature-fusion stage, a hyper-MFM structure was designed to address the problems of cross-scale semantic fragmentation and feature loss. Experimental results showed that compared with the RT-DETR baseline, the accuracy, mAP@50, and mAP@50:95 metrics were improved by 8.4%, 5.8%, and 4.8%, respectively, when using the MPH-DETR algorithm on the HIT-UAV infrared dataset, which effectively alleviated the problems of false detection and missed detection in complex backgrounds.

     

/

返回文章
返回