Synthetic Aperture Radar (SAR) target detection suffers from complex environments and small target sizes. The number of baseline model RT-DETR parameters is high. Aiming at the above problems, this paper proposes a lightweight small target detection in SAR images based on improved RT-DETR (LSTRT-DETR). Firstly, a lightweight additive convolutional GLU backbone network (LACGBNet) is designed to capture remote dependencies and optimize the feature extraction process to cope with complex scenes while achieving model lightweighting. Secondly, the algorithm proposes a small target feature fusion network (STFFN), which introduces SPD-Conv to process the P2 feature layer to enrich the small target feature information. And the shallow feature fusion module (SDFM) is introduced to enhance the detection capability of small targets. Experimental results on the joint dataset consisting of SSDD, HRSID, and SARAIRcraft show that the mAP50 and mAP50-95 of LSTRT-DETR reach 92.5% and 63.8%, respectively, which are improved by 0.4% and 3%, respectively, and the number of parameters is reduced by 4.1 M compared to the original model RT-DETR. This sufficiently demonstrates its effectiveness and efficiency in the SAR image target detection task.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Lightweight Small Target Detection in SAR Images Based on Improved RT-DETR

  • Tanqing Sun,
  • Xianjun Zhang,
  • Ziyu Li,
  • Na You

摘要

Synthetic Aperture Radar (SAR) target detection suffers from complex environments and small target sizes. The number of baseline model RT-DETR parameters is high. Aiming at the above problems, this paper proposes a lightweight small target detection in SAR images based on improved RT-DETR (LSTRT-DETR). Firstly, a lightweight additive convolutional GLU backbone network (LACGBNet) is designed to capture remote dependencies and optimize the feature extraction process to cope with complex scenes while achieving model lightweighting. Secondly, the algorithm proposes a small target feature fusion network (STFFN), which introduces SPD-Conv to process the P2 feature layer to enrich the small target feature information. And the shallow feature fusion module (SDFM) is introduced to enhance the detection capability of small targets. Experimental results on the joint dataset consisting of SSDD, HRSID, and SARAIRcraft show that the mAP50 and mAP50-95 of LSTRT-DETR reach 92.5% and 63.8%, respectively, which are improved by 0.4% and 3%, respectively, and the number of parameters is reduced by 4.1 M compared to the original model RT-DETR. This sufficiently demonstrates its effectiveness and efficiency in the SAR image target detection task.