用于野外机器人的红外小目标检测算法

Infrared Small Target Detection Algorithm for Field Robots

  • 摘要: 红外热成像目标检测对机器人在野外环境实现全天候巡检有重要意义。针对机器人机载的嵌入式计算机运算能力不足、目标检测实时性低以及红外热成像图形的小目标分辨率低的问题。提出了一种基于YOLOv7改进的轻量级检测算法。首先,对网络进行结构化剪枝以提高在嵌入式计算机上运行的实时性,再构建自适应卷积层和无批次归一化处理模块改进骨干网络。然后,使用多速率空洞卷积与3D卷积提取高分辨率的尺度序列特征,再使用FPN将尺度序列特征做特征融合,增强小目标检测精度。最后,在预测阶段引入SIoU位置回归方法,提升回归速度和准确度。通过NVIDIA Jetson Xavier NX平台在夜间红外热成像数据集上验证表明,与YOLOv7相比,以mAP降低1.95%为代价换取了FPS 162%的提升,达到实时性检测的要求。

     

    Abstract: Infrared (IR) thermal imaging target detection is essential for enabling robots to conduct all-weather inspections in field environments. This paper addresses two key challenges: the limited computing power of embedded systems onboard robots for real-time detection, and the low resolution of small targets in thermal imaging. To address these challenges, a lightweight detection algorithm based on an improved YOLOv7 framework is proposed. First, the network structure is pruned to enhance real-time performance on embedded devices. Subsequently, the backbone is optimized by integrating adaptive convolutional layers and a batchless normalization module. To improve small-target detection accuracy, multi-rate dilated 3D convolution is used to extract high-resolution scale-sequence features, which are subsequently fused via a Feature Pyramid Network (FPN). Finally, the SIoU-based position regression method is introduced in the prediction stage to improve regression speed and accuracy. Experimental validation on the NVIDIA Jetson Xavier NX platform using a nighttime thermal imaging dataset shows a 162% improvement in FPS, with only a 1.95% reduction in mAP compared to the original YOLOv7, meeting the requirements for real-time detection.

     

/

返回文章
返回