基于自注意力机制的点云分类分割

作者：孟繁林; 何晓曦^*; 刘应浒; 李茄濡; 朱群

来源：计算机系统应用, 2023, 1-8.

DOI：10.15888/j.cnki.csa.009341

• 深度学习
• 自注意力机制
• 分类
• 分割
• 点云

3D点云由于其无序性以及缺少拓扑信息使得点云的分类与分割仍具有挑战性.针对上述问题,我们设计了一种基于自注意力机制的3D点云分类算法,可学习点云的特征信息用于目标分类与分割.首先,设计适用于点云的自注意力模块,用于点云的特征提取.通过构建领域图来加强输入嵌入,使用自注意力机制进行局部特征的提取与聚合.最后,通过多层感知机以及解码器-编码器的方式将局部特征进行结合,实现3D点云的分类与分割.该方法考虑了输入嵌入时单个点在点云中的局部语境信息,构建局部长距离下的网络结构,最终得到的结果更具区分度.在ShapeNetPart、RoofN3D等数据集上的实验证实文中方法的分类与分割性能较优.

• 单位
  成都信息工程大学