基于智能半自动标注的铁路LiDAR语义分割方法
Railway LiDAR semantic segmentation based on intelligent semi-automated data annotation
- 融合图像与点云,用2DPass网络实现点云语义分割
- 在9类场景下实现71.48%的平均交并比
- 结合主动学习提升小样本标注效率,适合铁路自动驾驶
自动驾驶车辆依赖对环境的精准感知。与自动驾驶汽车类似,高度自动化列车也需要环境感知能力。尽管汽车领域已有大量基于摄像头或激光雷达(LiDAR)的研究,但针对自动化列车的相关工作仍十分有限,且目前尚无公开的铁路环境3D LiDAR语义分割数据集或方法。为此,本文提出一种基于2DPass网络架构的点云级3D语义分割方法,联合使用扫描数据与图像信息。同时,设计了一种智能半自动数据标注方案,高效准确地标注德国铁路轨道上采集的数据。为应对标注样本仍较少的问题,采用主动学习策略,智能筛选用于训练的点云扫描。主要贡献包括:构建包含相机与LiDAR数据的铁路数据集;利用图像分割网络对原始点云进行标签迁移;通过主动学习高效训练前沿3D LiDAR语义分割网络。最终模型在9个类别上达到71.48%的平均交并比。
原文摘要 · Abstract (English)
Automated vehicles rely on an accurate and robust perception of the environment. Similarly to automated cars, highly automated trains require an environmental perception. Although there is a lot of research based on either camera or LiDAR sensors in the automotive domain, very few contributions for this task exist yet for automated trains. Additionally, no public dataset or described approach for a 3D LiDAR semantic segmentation in the railway environment exists yet. Thus, we propose an approach for a point-wise 3D semantic segmentation based on the 2DPass network architecture using scans and images jointly. In addition, we present a semi-automated intelligent data annotation approach, which we use to efficiently and accurately label the required dataset recorded on a railway track in Germany. To improve performance despite a still small number of labeled scans, we apply an active learning approach to intelligently select scans for the training dataset. Our contributions are threefold: We annotate rail data including camera and LiDAR data from the railway environment, transfer label the raw LiDAR point clouds using an image segmentation network, and train a state-of-the-art 3D LiDAR semantic segmentation network efficiently leveraging active learning. The trained network achieves good segmentation results with a mean IoU of 71.48% of 9 classes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。