Real-time segmentation of various insulators using generative adversarial networks

被引:26
作者
Chang, Wenkai [1 ,2 ]
Yang, Guodong [1 ]
Yu, Junzhi [1 ]
Liang, Zize [1 ]
机构
[1] Chinese Acad Sci, Inst Automat, State Key Lab Management & Control Complex Syst, Beijing 100190, Peoples R China
[2] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
基金
中国国家自然科学基金;
关键词
image segmentation; insulators; neural nets; power engineering computing; real-time pixel-level segmentation; generative adversarial networks; insulator segmentation algorithm; cluttered background; artificial thresholds; compact end-to-end neural network; visual saliency map; proposed two-stage training; segmentation quality; ACTIVE CONTOUR MODEL;
D O I
10.1049/iet-cvi.2017.0591
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The conventional inspection of fragile insulators is critical to grid operation and insulator segmentation is the basis of inspection. However, the segmentation of various insulators is still difficult because of the great differences in colour and shape, as well as the cluttered background. Traditional insulator segmentation algorithms need many artificial thresholds, thereby limiting the adaptability of algorithms. A compact end-to-end neural network, which is trained in the framework of conditional generative adversarial networks, is proposed for the real-time pixel-level segmentation of insulators. The input image is mapped to a visual saliency map, and various insulators with different poses are filtered out at the same time. The proposed two-stage training and empty samples are also used to improve the segmentation quality. Extensive experiments and comparisons are performed on many real-world images. The experimental results demonstrate superior segmentation and real-time performance. Meanwhile, the effectiveness of the proposed training strategies and the trade-off between performance and speed are analysed in detail.
引用
收藏
页码:596 / 602
页数:7
相关论文
共 35 条
[1]  
Badrinarayanan V., 2016, ARXIV151100561
[2]  
Chandra S., 2016, ARXIV160308358V1
[3]  
Chang WK, 2017, IEEE INT C INT ROBOT, P973, DOI 10.1109/IROS.2017.8202263
[4]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[5]  
Chetlur S, 2014, ARXIV
[6]  
Han Y, 2016, IEEE IMAGE PROC, P3852, DOI 10.1109/ICIP.2016.7533081
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]  
Ioffe Sergey, 2015, P MACHINE LEARNING R, V37, P448, DOI [DOI 10.48550/ARXIV.1502.03167, DOI 10.5555/3015118.3045167]
[9]  
Isola P, 2017, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, P1125, DOI [DOI 10.1109/CVPR.2017.632, 10.1109/CVPR.2017.632]
[10]  
Jabid T, 2016, P INT C INN SCI ENG, P1, DOI DOI 10.1109/ICISET.2016.7856522