基于改进DGCNN的点云分类方法研究

RESEARCH ON POINT CLOUD CLASSIFICATION BASED ON IMPROVED DGCNN

  • 摘要: 为了加强DGCNN局部图中点的交互能力,设计一种新的向下过渡层来替换DGCNN中的EdgeConv模块来完成分类任务。在向下过渡层中使用切向量来增加点的几何特征表达,通过FPS和KNN算法进行分组和采样,将采样分组后的点云输入到几何仿射模块GAM和MLP中完成局部特征提取。设计一种基于密度相关统计的共享边缘函数和基于K-Farthest算法的局部分组方法。在ModelNet40、ScanObjectNN和ShapeNet数据集上进行分类实验,该方法在与DGCNN的结果对比中均取得了较好的效果。

     

    Abstract: In order to enhance the interaction of points in DGCNN local graphs, the paper designs a new transition down layer to replace the EdgeConv module in DGCNN to accomplish the classification task. This paper used tangent vectors in the transition down layer to increase the geometric feature representation of points, grouped and sampled them by FPS and KNN algorithms, and inputted the sampled and grouped point clouds into the geometric affine modules GAM and MLP to complete the local feature extraction. This paper designed a shared edge function based on density-dependent statistics and a local grouping method based on K-Farthest algorithm. Classification experiments were conducted on ModelNet40, ScanObjectNN and ShapeNet datasets, and the proposed method achieved better results in all comparisons compared with DGCNN results.

     

/

返回文章
返回