基于局部特征聚合与驱动点云采样的点云分类与分割

Point cloud classification and segmentation based on local feature aggregation and driving point cloud sampling

  • 摘要: 针对目前大多数深度学习模型获取局部特征的能力比较欠缺的问题,提出了一种基于局部特征聚合与驱动点云采样的点云分类与分割网络——FCA-PointNet++。设计了局部特征聚合模块,在该模块中构建了增强相关局部特征并抑制不相关特征的FCA(feature channel attention)模块;设计了一种根据不同任务改变初始采样点分布的驱动点云采样网络(drive point cloud sampling,DPN);构造了一种能准确提取局部特征并且自适应细化特征的通道门(channel-wise gate)模块。在分类任务中,FCA-PointNet++在ModelNet10和ModelNet40上的总体准确率(overall accuracy,OA)分别达到了95.3%、93.4%。在分割任务中,FCA-PointNet++的平均交并比(mean intersection over union,MIoU)达到了86.1%,实验验证了FCA-PointNet++在分类和分割任务中的有效性。

     

    Abstract: Aiming at the problem that most deep learning models lack the ability to obtain local features, a point cloud classification and segmentation network (FCA-Pointnet ++) based on local feature aggregation and driven point cloud sampling was proposed. A local feature aggregation module was designed and a Feature channel attention (FCA) module was constructed to enhance relevant local features and suppress irrelevant features. A driver point cloud sampling network (DPN) which could change the distribution of initial sampling points according to different tasks was designed. A channel-wise gate module, which could extract local features more accurately and refine features adaptively, was constructed. In the classification task, FCA-Pointnet ++ achieved an overall accuracy (OA) of 95.3% and 93.4% on ModelNet10 and ModelNet40, respectively. In the segmentation task, the mean intersection over Union(mIoU) of FCA-Pointnet ++ reached 86.1%, and the effectiveness of FCA-Pointnet ++ in classification and segmentation tasks was verified by experiments.

     

/

返回文章
返回