NFZ-YOLOv10:面向机场管制区的红外目标检测算法

NFZ-YOLOv10: Infrared Target Detection Algorithm for Airport Control Zones

  • 摘要: 针对红外场景下目标检测模型计算复杂、小目标检测效果不佳等问题,提出了一种改进的目标检测算法NFZ-YOLOv10。通过引入轻量化网络StarNet,降低模型的计算量和参数量,同时提升模型的特征提取能力;优化颈部网络,利用GSConv模块和VOVGSCSP模块构建Slim-Neck范式,实现对深层次语义信息与细粒度空间特征的深度融合;引入NWD损失函数,有效缓解传统损失函数对小尺度目标检测任务中位置偏差敏感的问题,显著提升模型的检测精度。实验结果表明,NFZ-YOLOv10在自建数据集上的平均精度均值达到94.5%,较YOLOv10n提升2.7%。此外,NFZ-YOLOv10的运算量和参数量相较于YOLOv10n分别降低了24.4%与22.3%,单帧图像检测时间低至4.5 ms,模型体积缩小至4.9 MB。NFZ-YOLOv10在检测精度与复杂度之间实现了更为有效的平衡,在机场管制区内对于低慢小目标的检测方面具有较强的应用潜力。

     

    Abstract: To address the high computational complexity and poor detection performance of small targets in infrared scenarios, an improved target-detection algorithm, NFZ-YOLOv10, is proposed. By introducing the lightweight network StarNet, the computational load and parameter count of the model were reduced, and the feature extraction capability was enhanced. The neck network was optimized, and the slim-neck paradigm was constructed using the GSConv and VOVGSCSP modules, achieving a deep integration of high-level semantic information and fine-grained spatial features. The NWD loss function was introduced to effectively alleviate the sensitivity of traditional loss functions to positional deviations in small-scale target-detection tasks, thereby significantly improving the detection accuracy of the model. The experimental results showed that the mean average precision of NFZ-YOLOv10 on the self-built dataset reached 94.5%, an increase of 2.7% compared to YOLOv10n. Additionally, the computational cost and the number of parameters of NFZ-YOLOv10 have been reduced by 24.4% and 22.3%, respectively, compared to YOLOv10n. The single-frame image detection time is as low as 4.5 ms, and the model size has been shrunk to 4.9 MB. In summary, NFZ-YOLOv10 achieves a more effective balance between detection accuracy and complexity and has strong application potential in the detection of low-slow-small targets within airport control zones.

     

/

返回文章
返回