改进YOLOv7的电力设备红外图像多目标检测

Improved YOLOv7 for Multi-Target Detection of Infrared Images of Power Equipment

  • 摘要: 针对变电站红外图像中目标数量多、外观相似、背景与目标颜色相近时容易出现漏检误检等问题,提出了一种改进YOLOv7的电力设备红外图像多目标检测方法。首先,为了更好地保留红外图像中的浅层信息,在SPPCSPC模块中引入空洞卷积与均值池化以扩大感受野的同时防止红外小目标淹没在背景中;其次,针对多目标检测中的误检漏检等问题,在头部网络中引入了轻量型SimAM注意力机制以重点关注感兴趣区域。最后,选择一个更适用于小目标检测的NWDloss损失与CIOU损失相结合的混合边框回归损失函数,它可以有效提高红外图像中不同尺度目标检测的准确性。我们在自建电力设备红外图像数据集上与其他7种具有代表性的检测方法进行了对比实验。实验结果表明,改进后的YOLO v7网络模型漏检误检等情况得到明显改善,mAP达到88.9%,相比其他有代表性的目标检测算法在电力设备红外多目标检测上的效果有明显提升。

     

    Abstract: To address issues such as a large number of targets, similar appearance, and easy to miss detection and misdetection when the background and target are of similar color in substation infrared images, a multi-target detection method is proposed for electric power equipment by improving YOLOv7. First, to better retain the shallow information in the infrared image, cavity convolution and mean pooling were introduced into a spatial pyramid pooling cross-stage partial convolution (SPPCSPC) module, to expand the receptive field while preventing small infrared targets from being submerged in the background. Second, to deal with misdetection and detection omission in multi-target detection, a lightweight simple attention module (SimAM) was introduced into the head network to focus on the region of interest. Finally, a hybrid edge regression loss function suitable for small-target detection, combining the normalized Gaussian Wasserstein distance (NWD) and complete intersection over union (CIOU) losses, was chosen to effectively improve the accuracy of target detection at different scales in infrared images. We conducted comparison experiments with seven other representative detection methods using a self-constructed infrared image dataset of power equipment. The experimental results showed that the improved YOLOv7 network model significantly improves leakage detection and reduced false detection. Its mean average precision (mAP) reached 88.9%, which is a significant improvement compared with those of other representative target detection algorithms for infrared multi-target detection of power equipment.

     

/

返回文章
返回