Welcome to Metal Mine! Today is Share:
×

扫码分享

Metal Mine ›› 2023, Vol. 52 ›› Issue (05): 237-246.

Previous Articles     Next Articles

Subway Tunnel Point Cloud Semantic Segmentation Network Model Based on Spatial Geometric Feature Fusion

ZHANG Qiuzhao1,2 LIANG Jiahui1 DUAN Haoran1 WANG Zongwei2,3 DUAN Wei1,4 #br#   

  1. 1. School of Environment and Spatial Informatics,China University of Mining and Technology,Xuzhou 221116,China;2. Key Laboratory of Land Satellite Remote Sensing Application,Ministry of Natural Resources,Nanjing 210013,China;3. Jiangsu Province Surveying & Mapping Engineering Institute,Nanjing 210013,China;4. Nanjing Institute of Surveying,Mapping & Geotechnical Investigation,Co. . Ltd. ,Nanjing 210019,China
  • Online:2023-05-15 Published:2023-06-15

Abstract: The tunnel point cloud semantic segmentation technology based on deep learning can recognize and classify objects in large-scale point cloud data,and realize the extraction and management of object information in tunnel scenes. A subway tunnel point cloud semantic segmentation network model based on enhanced spatial geometric feature fusion is proposed in this paper,which takes into account the spatial distribution characteristics and geometric features of subway tunnel facilities. A spatial geometric feature extraction module is designed to extract the relative spatial positions and geometric distribution features of tunnel facilities and combines them with the features of the network encoding layer through channel concatenation,enhancing the network′s perception of multi-scale feature information. A feature fusion encoding layer based on channel attention mechanism is designed to extract the weight of features between different channels,and to perform weighted fusion of the information from different spatial scales,so as to fully utilize information of different scales to improve model representation and generalization abilities. The network model proposed in this paper is verified using a dataset made by the point cloud data of a tunnel in Nanjing Subway. The experimental results demonstrate that the proposed network achieved a training mIoU value of 0. 955 6. The predicted results on the test dataset showed a weighted average F1 score of 0. 995 9 and a weighted average IoUvalue of 0. 963 1. For the communication cables,pipeline racks,and contact hangers classes with poor IoU values in the PointNet++ network,the IoU values of the proposed network reached 0. 845,0. 825,and 0. 999,respectively,which effectively improved the overall accuracy of point cloud segmentation of subway tunnel and provided a technical reference for realizing automatic disease inspection and facility management of subway tunnel.

Key words: point cloud semantic segmentation,channel attention mechanism,spatial geometric feature,PointNet++