CycleGAN-Based Data Augmentation for Subgrade Disease Detection in GPR Images with YOLOv5

被引:4
作者
Yang, Yang [1 ]
Huang, Limin [1 ]
Zhang, Zhihou [1 ]
Zhang, Jian [1 ]
Zhao, Guangmao [2 ]
机构
[1] Southwest Jiaotong Univ, Fac Geosci & Environm Engn, Chengdu 611756, Peoples R China
[2] China Railway Design Corp, Tianjin 300142, Peoples R China
关键词
disease detection; ground-penetrating radar; YOLO; GAN; semi-supervised learning; DEEP NEURAL-NETWORKS; CLASSIFICATION; RECOGNITION;
D O I
10.3390/electronics13050830
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Vehicle-mounted ground-penetrating radar (GPR) technology is an effective means of detecting railway subgrade diseases. However, existing methods of GPR data interpretation largely rely on manual identification, which is not only inefficient but also highly subjective. This paper proposes a semi-supervised deep learning method to identify railway subgrade diseases. This method addresses the sample imbalance problem in the defect dataset by utilizing a data augmentation method based on a generative adversarial network model. An initial network model for disease identification is obtained by training the YOLOv5 network with a small number of existing samples. The intelligently extended samples are then labeled to achieve a balance in the disease samples. The network is trained to improve the recognition accuracy of the intelligent model using a more complete dataset. The experimental results show that the accuracy of the proposed method can reach up to 94.53%, which is 23.85% higher than that of the supervised learning model without an extended dataset. This has strong industrial application value for railway subgrade disease detection as the potential learning ability of the model can be explored to a greater extent, thereby improving the recognition accuracy of subgrade diseases.
引用
收藏
页数:18
相关论文
共 48 条
[1]  
[Anonymous], 2007, P 15 ACM INT C MULT, DOI DOI 10.1145/1291233.1291430
[2]   Non-destructive Assessment and Health Monitoring of Railway Infrastructures [J].
Artagan, Salih Serkan ;
Ciampoli, Luca Bianchini ;
D'Amico, Fabrizio ;
Calvi, Alessandro ;
Tosti, Fabio .
SURVEYS IN GEOPHYSICS, 2020, 41 (03) :447-483
[3]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, 10.48550/arXiv.2004.10934, DOI 10.48550/ARXIV.2004.10934]
[4]   An algorithm for automatic localization and detection of rebars from GPR data of concrete bridge decks [J].
Dinh, Kien ;
Gucunski, Nenad ;
Duong, Trung H. .
AUTOMATION IN CONSTRUCTION, 2018, 89 :292-298
[5]   Special Issue on Ground Penetrating Radar: Theory, Methods, and Applications [J].
Feng, Deshan ;
Liu, Yuxin ;
Zhang, Bin ;
Wang, Xun .
APPLIED SCIENCES-BASEL, 2023, 13 (17)
[6]   Audio-visual human recognition using semi-supervised spectral learning and hidden Markov models [J].
Feng, Wei ;
Xie, Lei ;
Zeng, Jia ;
Liu, Zhi-Qiang .
JOURNAL OF VISUAL LANGUAGES AND COMPUTING, 2009, 20 (03) :188-195
[7]   A Deep Convolutional Generative Adversarial Networks (DCGANs)-Based Semi-Supervised Method for Object Recognition in Synthetic Aperture Radar (SAR) Images [J].
Gao, Fei ;
Yang, Yue ;
Wang, Jun ;
Sun, Jinping ;
Yang, Erfu ;
Zhou, Huiyu .
REMOTE SENSING, 2018, 10 (06)
[8]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[9]  
He J., 2004, MULTIMEDIA '04, P9, DOI DOI 10.1145/1027527.1027531
[10]   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