国内刊号:11-2436/TN
国际刊号:1001-5078
发布日期:
作者:陈科
单位:1.四川轻化工大学 自动化与信息工程学院,四川 宜宾 644000;2.智能感知与控制四川省重点实验室,四川 宜宾 644000;
关键词:红外图像;小目标检测;RTDETR;轻量化;Transformer
基金:四川省科技计划项目(No.2024YFHZ0026)资助。
针对无人机红外图像中小目标因信号微弱、背景复杂和尺寸小导致检测困难的问题,论文提出一种轻量级高精度检测模型 ERMR DETR。该模型基于 RT DETR R18,融合多项结构改进:首先引入GhostNet与RepVGG构建的轻量特征提取模块GhostRepCSP,提升表达能力并减少参数;然后在颈部网络中加入Retentive注意机制构建的特征融合模块RetBlockC3,增强上下文感知;最后通过设计多路径鲁棒下采样结构RFD,有效保留小目标纹理信息。实验在HIT UAV数据集上验证结果表明,提出模型mAP@05提升至956,mAP@05:095达611,而参数仅为122 M,表现出优异的精度与效率。在VisDrone2019与ALU数据集上的泛化实验,也验证了其良好的跨场景适应性与实际部署潜力。
来源:2025年第12期
《激光与红外》期刊编辑部