Rock mass structural surface trace extraction based on transfer learning

To solve engineering geological problems, including water conservancy, transportation, and mining, it is necessary to obtain information on rock mass structures, such as slopes, foundation pits, and tunnels, in time. The traditional method for obtaining structural information requires manual measurement, which is time consuming and labor intensive. Because geological information is complicated and diverse, it is not practical for general deep learning methods to obtain full-scale structural surface trace images to prepare training samples. Transfer learning can abstract high-level features from low-level features with a small number of training samples, which can automatically express the inherent characteristics of objects. This article proposed a rock mass structural surface trace extraction method based on the transfer learning technique that considers the attention mechanism and shape constraints. For the general test set, the accuracy of rock mass structural surface trace recognition with the proposed method can reach 87.2%. Experimental results showed that the proposed method has advantages in extracting complicated geological structure information and is valuable for providing technical support for the extraction of geological information in the construction of water conservancy, transportation, mining, and related projects.