基于改进YOLOv5的车辆红外图像多目标识别方法

Multi-target Recognition Method for Vehicle Infrared Images Based on Improved YOLOv5

城乡结合部的建设是城市建设中重要的一环,由于难以布设有效的检测设备,该区域车辆目标的夜间监管一直是城市管理的难题;基于无人机平台红外夜视图像多运动目标检测为解决这一难题提供了智能化路径:一种基于改进YOLOv5的红外夜视条件下多运动目标识别方法,分析了交通对象特征、车辆停放对道路红外辐射影响等,引入了CBAM注意力机制,以提取和融合空间和通道信息,增强了网络对目标的表达能力;综合Efficient IOU Loss和Focal Loss的优点,使用EIoU-Focal Loss损失函数替换了CIoU Loss函数,解决了样本不平衡、红外图像的低分辨率、噪声干扰大以及目标与背景对比度低等弊端,提高了检测的准确性;通过加入DCN动态调整卷积核的形状,适应图像中目标的形变,降低因形状不规则、变化较多造成的识别影响;在公开数据集上对改进网络与经典网络进行实验和数据比较,结果表明:综合改进后的网络对于多目标的识别,在YOLOv5x网络较高的识别结果基础上,精度提升3.9%,召回率提升4.1%,F_(1)增加4.4%。

The rural-urban fringe is an important part of urban construction.it is difficult to effectively deploy detection equipment,the night supervision of vehicle targets in this area always is a difficult problem for urban management.This paper provides an intelligent approach of detecting multiple moving targets for solving the problem in infrared night vision images based on UAV platforms,presents a multi-moving target recognition method based on improved YOLOv5 in infrared night vision conditions,and analyzes the characteristics of traffic objects and impact of vehicle parking on road infrared radiation,etc.Convolutional block attention module(CBAM) attention mechanism is introduced to extract and integrate spatial with channel information to enhance the expression ability of the network on the target.By combining the advantages of efficient IOU loss and focal loss,the EIoU-focal loss function is used to replace the CIoU loss function,solve the disadvantages of sample imbalance,low resolution of infrared image,large noise interference and low contrast between target and background,and improve the detection accuracy.By adding the DCN to dynamically adjust the shape of the convolution kernel,it can adapt to the deformation of the object in images,and reduce the recognition influence caused by irregular shape and many changes.Finally,experiments and data comparisons between improved network and typical network are implemented on public dataset,the results show that for multi targets recognition,the improved network based on YOLOv5 has higher recognition results,it increases the accuracy,recall rate and F_(1) value by 3.9%,4.1% and 4.4%,respectively.