• 基于三维点云的列车表面缺陷快速检测算法

    Rapid detection algorithm for train surface defects based on 3D point clouds

    • 针对列车车体表面缺陷点云检测中计算复杂度与检测精度难以兼顾的问题,提出一种融合局部几何特征增强与知识蒸馏的轻量化点云分割方法(PointNet-LSD)。以 PointNet 为基础骨干网络,在特征提取阶段构建基于K近邻的局部几何特征聚合模块,通过显式建模点与邻域间的相对空间关系,增强模型对细粒度几何结构的表达能力;同时引入通道注意力机制,对特征通道进行自适应重标定,以抑制大面积平整背景带来的噪声干扰。在模型优化阶段,采用教师-学生知识蒸馏策略,以 PointNet++ 作为教师网络,通过软标签分布约束引导学生模型学习更具判别性的特征表示,从而提升轻量化模型的泛化能力。在自建列车车体点云数据集上的实验结果表明,所提方法在参数量仅为1.67×106的条件下,实现了378.59 FPS 的推理速度;与基准 PointNet 模型相比,交并比、准确率和F1值分别提升约5.9%、4.5%和2.9%。结果表明,该方法在保证计算效率的同时,能够有效提升对微小缺陷的检测性能,具有一定的工程应用价值。

       

      Abstract: To address the challenge of balancing computational complexity and detection accuracy in point cloud-based defect detection for train body surfaces, a lightweight point cloud segmentation method, named PointNet-LSD, is proposed. Based on the PointNet backbone, a local geometric feature aggregation module is introduced in the feature extraction stage, where K-nearest neighbors are utilized to explicitly model the spatial relationships between points and their neighborhoods, thereby enhancing the representation capability for fine-grained geometric structures. Meanwhile, a channel attention mechanism is incorporated to adaptively recalibrate feature channels and suppress noise interference from large-area smooth backgrounds. In the model optimization stage, a teacher–student knowledge distillation strategy is adopted, where PointNet++ serves as the teacher network, and soft label distributions are used to guide the student network to learn more discriminative feature representations, thus improving the generalization ability of the lightweight model. Experimental results on a self-constructed train body point cloud dataset demonstrate that the proposed method achieves an inference speed of 378.59 FPS with only 1.67×106parameters. Compared with the baseline PointNet model, the intersection over union, accuracy, and F1-score are improved by approximately 5.9%, 4.5%, and 2.9%, respectively. The results indicate that the proposed method can effectively enhance the detection performance of small defects while maintaining high computational efficiency, showing promising potential for practical engineering applications.

       

    /

    返回文章
    返回