轻量化红外道路小目标实时检测算法

Lightweight Real-Time Detection Algorithm for Infrared Small Road Targets

  • 摘要: 近年来,红外目标检测在智能交通、自动驾驶领域发挥了至关重要的作用。现有的红外目标检测技术在处理低对比度、检测小目标、保证实时性等方面存在困难。本文在YOLOv8算法的基础上提出了一种名为KCL-YOLOv8s的红外道路目标检测算法,该算法设计了一种名为C2f_LSK的模块替代原算法中的C2f模块,增强了网络多尺度特征提取能力。此外,还设计了一种名为还设计了一种特征聚焦扩散金字塔网络(Feature-centric diffusion pyramid network,CDPN)的特征聚焦扩散金字塔网络,增强了模型对小目标的检测能力。同时,考虑到车载计算资源有限的问题,设计了一种轻量化共享卷积头,提升模型检测效率。最后,提出了WIoU v3的改进版本WIoU v3*作为损失函数替代原模型中的损失函数,提高了模型分类和定位目标的能力,同时提升了模型对红外小目标的检测能力。实验结果表明,改进后的模型在FLIR_ADAS_v2数据集上的平均检测精度mAP0.5为70.3%,优于其他先进模型。与YOLOv8s模型相比,其mAP0.5提高了3.8%,参数量减少了26.2%,模型处理速度提升了35.8%,更适合部署在需要实时性且资源受限的场景中。

     

    Abstract: Recently, infrared object detection has played a crucial role in intelligent transportation and autonomous driving. Existing infrared object detection technologies face challenges in handling low-contrast scenes, detecting small objects, and ensuring real-time performance. KCL-YOLOv8s, an infrared road object detection algorithm based on the YOLOv8 framework, was proposed in this study. This algorithm introduced a module named C2f_LSK to replace the C2f module in the original algorithm, thereby enhancing the network's multiscale feature extraction capabilities. In addition, a feature-centric diffusion pyramid network (CDPN) was designed to enhance the ability of the model to detect small objects. Considering the limited computational resources available for vehicle-mounted systems, a lightweight shared convolutional head was designed to improve the detection efficiency of the model. Finally, an improved version of WIoU v3, termed WIoU v3*, was adopted as the loss function to replace the one in the original model, thereby improving the model's ability to classify and localize objects and enhancing its detection capability for small infrared targets. Experimental results showed that the improved model achieved a mean average precision (mAP0.5) of 70.3% on the FLIR_ADAS_v2 dataset, outperforming other state-of-the-art models. Compared with the YOLOv8s model, it achieved a 3.8% increase in mAP0.5, 26.2% reduction in the number of parameters, and 35.8% improvement in processing speed, making it more suitable for deployment in real-time, resource-constrained scenarios.

     

/

返回文章
返回