基于改进YOLOv5模型的车载红外目标检测

Vehicle-mounted infrared target detection based on improved YOLOv5 model

  • 摘要: 针对车载红外目标检测算法中,因红外小目标特征少、对比度低导致的检测精度低的问题,提出了FE-YOLOv5s车载红外目标检测模型。首先,在主干网络中引入PPA模块以弥补下采样操作带来的特征信息损失;其次,基于AFPN网络和DASI模块,构建特征融合网络DAFPN,直接融合非临近尺度层的特征信息,同时将五种不同尺度层的特征信息都纳入最终的特征输出,进一步增强目标的特征表达能力,并重构目标预测层,以提高模型对小目标的识别率;最后,采用MPDIoU作为边界框损失函数以提升目标的定位精度。实验结果表明,本文所提出的模型在FLIR数据集和艾睿车载红外数据集上的mAP@0.5分别达到了88.1%和84.7%,相比YOLOv5s提升了7.2%和3.2%,缓解了模型对红外小目标错检漏检的问题,且检测速度为100 帧/s,为嵌入式部署提供了可行性。

     

    Abstract: To address the issue of low detection accuracy in vehicle-mounted infrared target detection algorithms, caused by the limited features and low contrast of small targets, this paper proposed the FE-YOLOv5s vehicle-mounted infrared target detection model. Firstly, the PPA module was integrated into the backbone network to mitigate the feature information loss caused by downsampling operations. Secondly, based on the AFPN network and the DASI module, we developed a feature fusion network termed DAFPN. This network directly fuses feature information from non-adjacent scale layers and incorporates features from five different scale layers into the final feature output, further enhancing the feature representation capability of the targets. The target prediction layer was reconstructed to improve the model's recognition rate of small targets. Finally, MPDIoU was adopted as the bounding box loss function to enhance target localization accuracy. Experimental results demonstrated that the proposed model achieves mAP@0.5 scores of 88.1% on the FLIR dataset and 84.7% on the iRay vehicle-mounted infrared dataset, representing improvements of 7.2% and 3.2% over the YOLOv5s model. It effectively alleviates the problem of false and missed detections of infrared small targets while operating at 100 FPS, which confirms its potential for embedded deployment.

     

/

返回文章
返回