基于改进RCF和无人机影像的电力线检测

Power line detection based on an improved RCF and the UAV images

针对更丰富卷积特征(RCF)算法检测电力线时存在边缘模糊、特征图包含太多噪声、在融合特征图时丢失多尺度信息等问题,对RCF算法进行改进.首先,使用具有平移不变性的下采样技术增强模型的鲁棒性;然后,在RCF主干网络中引入卷积块注意力模块(CBAM)机制,提高模型对电力线特征的表达能力;最后,在RCF的侧输出网络中加入级联网络,借助基于通道注意力机制的多尺度特征融合模块对特征图进行融合,从而获得更优异的细节保持效果.实验结果表明,改进模型的最优数据集规模、最佳图像比例和平均精度可分别提高0.7%、 1.3%和1.7%,检测结果噪声数量少,电力线更加清晰准确.

In order to solve the problems such as edge blur when richer convolutional features(RCF)algorithm detects power lines,feature maps contain too much noise,and multi-scale information is lost when fusing feature maps,RCF algorithm was improved in this paper.Firstly,the down-sampling technique with translation invariance was used to enhance model robustness.Secondly,convolutional block attention module(CBAM)mechanism was introduced into the convolutional block attention of RCF trunk network to enhance the power lines characteristics.Thirdly,the cascade network is added into the side output network of RCF,and the feature map is fused with the multi-scale feature fusion module using the channel attention mechanism,to obtain better details.The results showed that the optimal dataset scale,the optimal image scale and average precision of the improved RCF increased 0.7%,1.3%and 1.7%,respectively.The detection results of the improved model are less noise,and the power line is more clear and accurate.