Applied Sciences (Sep 2024)
Research on Intelligent Recognition Method of Ground Penetrating Radar Images Based on SAHI
Abstract
Deep learning techniques have flourished in recent years and have shown great potential in ground-penetrating radar (GPR) data interpretation. However, obtaining sufficient training data is a great challenge. This paper proposes an intelligent recognition method based on slicing-aided hyper inference (SAHI) for GPR images. Firstly, for the problem of insufficient samples of GPR images with structural loose distresses, data augmentation is carried out based on deep convolutional generative adversarial networks (DCGAN). Since distress features occupy fewer pixels on the original image, to allow the model to pay greater attention to the distress features, it is necessary to crop the original images centered on the distress labeling boxes first, and then input the cropped images into the model for training. Then, the YOLOv5 model is used for distress detection and the SAHI framework is used in the training and inference stages. The experimental results show that the detection accuracy is improved by 5.3% after adding the DCGAN-generated images, which verifies the effectiveness of the DCGAN-generated images. The detection accuracy is improved by 10.8% after using the SAHI framework in the training and inference stages, which indicates that SAHI is a key part of improving detection performance, as it significantly improves the ability to recognize distress.
Keywords