Attention Model and Soft-NMS-Based Transmission Line Small Target Detection Method

被引:0
作者
Zhao Y. [1 ]
Tian S. [1 ]
Li Y. [1 ]
Luo L. [1 ]
Qi P. [1 ]
机构
[1] State Grid Qinghai Electric Power Company Overhauling Company, Xining
来源
Dianzi Keji Daxue Xuebao/Journal of the University of Electronic Science and Technology of China | 2023年 / 52卷 / 06期
关键词
attention mechanism; small target detection; soft-NMS; transmission line;
D O I
10.12178/1001-0548.2022290
中图分类号
学科分类号
摘要
In the defect detection of transmission lines, the bird's nest, plastic, rags and other suspended solids are mostly small targets. They have few pixels in the images and are easy to be disturbed by the background, which make the detection accuracy needs to be improved. In this paper, a new two-stage object detection algorithm is designed to improve the detection effect of bird nests and suspended solids in transmission lines. In order to improve the detection performance of small targets, the attention mechanism is integrated into the feature extraction network to learn more rich context information. In addition, in the detection network, a post-processing method based on softer non maximum suppression algorithm is designed to reduce the loss of small targets. Compared with the commonly used two-stage object detection algorithms, the proposed method improves the average accuracy of the two categories by about 4.7% and 5.9%, respectively, and has greater value in practical applications. © 2023 Univ. of Electronic Science and Technology of China. All rights reserved.
引用
收藏
页码:906 / 914
页数:8
相关论文
共 32 条
[1]  
SHU Y, CHEN W., Research and application of UHV power transmission in China, High Voltage, 3, 1, pp. 1-13, (2018)
[2]  
PANG Y, WANG T, ANWER R M, Et al., Efficient featurized image pyramid network for single shot detector, Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7336-7344, (2019)
[3]  
YANG X, LIU Q, YAN J, Et al., R3det: Refined single-stage detector with feature refinement for rotating object
[4]  
HU J, SHEN L, SUN G., Squeeze-and-excitation networks, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7132-7141, (2018)
[5]  
WOO S, PARK J, LEE J Y, Et al., Cbam: Convolutional block attention module, Proceedings of the European Conference on Computer Vision (ECCV), pp. 3-19, (2018)
[6]  
HU J, SHEN L, ALBANIE S, Et al., Gather-excite: Exploiting feature context in convolutional neural networks, Advances in Neural Information Processing Systems, 31, pp. 9401-9411, (2018)
[7]  
Zh AO Z, XU G, QI Y, Et al., Multi-patch deep features for power line insulator status classification from aerial images, 2016 International Joint Conference on Neural Networks (IJCNN), pp. 3187-3194, (2016)
[8]  
GAO Q, WU Y, LI Q., A faulted insulator identification algorithm of sparse difference-based deep belief network, Electronic Measuring Instrument, 53, 1, pp. 19-25, (2016)
[9]  
LIU Y, YONG J, LIU L, Et al., The method of insulator recognition based on deep learning, 2016 4th International Conference on Applied Robotics for the Power Industry (CARPI), pp. 1-5, (2016)
[10]  
WANG W, TIAN B, LIU Y, Et al., Study on the electrical devices detection in UAV images based on region based convolutional neural networks, Journal of Geo-information Science, 19, 2, pp. 256-263, (2017)