投审稿入口

矿区变电站电气设备红外图像目标检测的改进Faster R-CNN算法

Improved Faster R-CNN Algorithm for Infrared Image Object Detection of Electrical Equipment in Mining-area Substations

  • 摘要: 针对矿区变电站电气设备红外图像目标检测中热点目标边界模糊、温度梯度变化复杂以及目标尺寸差异大等问题,现有检测算法普遍存在识别精度不足、漏检率较高等缺陷。通过对比多种主流目标检测算法在矿区变电站电气设备红外图像数据集上的性能,发现融合特征金字塔结构的Faster R-CNN算法表现较优,但其对小型热点目标和温度异常区域的识别效果仍有提升空间。为此,提出了一种改进型Faster R-CNN算法。该算法首先在网络中引入空间金字塔池化结构以提取多尺度特征;其次优化特征融合方式,增强网络对不同尺度目标的感知能力;在候选框生成阶段,通过扩展锚框尺度及比例提高区域提议质量;最后改进分类与边界框回归损失函数,以提高检测精确度。试验结果表明:相较于原始Faster R-CNN算法,改进算法在矿区变电站电气设备红外图像数据集上的检测平均精度AP(0.5∶0.95)提高了2.7个百分点,AP(0.5)提高了4.3个百分点,对小尺寸热点目标的检测精度提升了1.8个百分点,反映出改进算法在矿区电气设备红外检测中具有一定的适用性。

     

    Abstract: To address the problems of blurred boundaries of hot-spot targets,complex temperature gradient variations,and large differences in target scales in infrared image object detection for electrical equipment in mining-area substations,existing detection algorithms generally have deficiencies such as insufficient recognition accuracy and high missed detection rates. By comparing the performance of several mainstream object detection algorithms on an infrared image dataset of electrical equipment in mining-area substations,it is found that the Faster R-CNN algorithm incorporating a feature pyramid structure performs relatively well,but its recognition performance of small hot-spot targets and temperature anomaly regions still has room for improvement. Therefore,an improved Faster R-CNN algorithm is proposed. First,a spatial pyramid pooling structure is introduced into the network to extract multi-scale features. Second,the feature fusion method is optimized to enhance the network′s perception of targets at different scales. In the region proposal generation stage,the quality of region proposals is improved by expanding anchor scales and aspect ratios. Finally,the classification and bounding box regression loss functions are improved to enhance detection accuracy. Experimental results show that,compared with the original Faster R-CNN algorithm,the improved algorithm achieves improvements of 2. 7 percentage points in AP(0. 5 ∶0. 95),4. 3 percentage points in AP(0. 5),and 1. 8 percentage points in the detection accuracy for small-sized hot-spot targets on the infrared image dataset of electrical equipment in mining-area substations,reflecting that the improved algorithm has certain applicability to infrared detection of mining-area electrical equipment.

     

/

返回文章
返回