[发明专利]基于注意力机制的3D点云分类方法、终端设备及存储介质

说明书

本公开提供一种基于注意力机制的3D点云分类方法、终端设备及计算机可读存储介质，其中，所述方法包括：创建注意力机制神经网络模型，所述注意力机制神经网络模型包括M层相连的注意力机制卷积网络层；将3D点云逐级输入至所述注意力机制卷积网络层中，得到3D点云中所有点的点类特征；以及，基于3D点云中所有点的点类特征对所述3D点云进行分类。本公开针对3D点云的无序等特性，创建基于注意力机制的神经网络模型，在学习过程中可以关注到3D点云中重要的点及其特征持续被神经网络关注，且能够考虑到到点云中边的属性，相较于普通神经网络而言点云的分类结果准确率更高。

技术领域

本公开涉及计算机视觉技术领域，尤其涉及一种基于注意力机制的3D点云分类方法、一种终端设备以及一种计算机可读存储介质。

背景技术

3D点云是3D对象表面几何构造的一种无序化表示，可通过3D深度相机等3D视觉技术采集获取，3D点云可在三维物体重建、逆向工程设计、无人驾驶、机器人等领域中应用。因此，在机器视觉领域，已经有2D向3D视觉发展的趋势，3D点云数据也正在迅速增长。

与人类视觉不同的是，如何教会机器准确地识别3D对象并进行3D点云的高效分类，是一项极具挑战的任务，虽然目前已提出利用神经网络对3D点云进行深度学习来完成对3D点云的分类工作，来提高点云分类的准确定，但由于点云中的点有坐标，除了考虑点的属性，还要考虑边的属性，采用普通图神经网络进行点云分类时，其准确率较低。

发明内容

本公开提供了一种基于注意力机制的3D点云分类方法、终端设备及计算机可读存储介质，以至少解决目前采用普通神经网络对3D点云分类准确率低等问题。

为实现上述目的，本公开提供一种基于注意力机制的3D点云分类方法，包括：

创建注意力机制神经网络模型，所述注意力机制神经网络模型包括M层相连的注意力机制卷积网络层；

将3D点云逐级输入至所述注意力机制卷积网络层中，得到3D点云中所有点的点类特征；以及，

基于3D点云中所有点的点类特征对所述3D点云进行分类。

在一种实施方式中，所述方法还包括：

在所述注意力机制卷积网络层中插入残差连接。

在一种实施方式中，在将3D点云逐级输入至所述注意力机制卷积网络层中之前，还包括：

对3D点云进行对称变换，得到经过对称变换的3D点云；

将3D点云逐级输入至所述注意力机制卷积网络层中，得到所述3D点云中所有点的点类特征，包括：

将经过对称变换的3D点云逐级输入至所述注意力机制卷积网络层中，得到所述3D点云中所有点的点类特征。

在一种实施方式中，对3D点云进行对称变换，包括：

采用PointNet对称函数对3D点云进行对称变换。

在一种实施方式中，将3D点云逐级输入至所述注意力机制卷积网络层中，得到3D点云中所有点的点类特征，包括：

将3D点云逐级输入到第1层的注意力机制卷积网络层至第M层的注意力机制卷积网络层中，得到3D点云中所有点最终的全局点类特征和局部边缘点类特征；

对3D点云中所有点最终的全局点类特征和局部边缘点类特征进行池化聚合操作，得到3D点云中所有点的点类特征。

在一种实施方式中，所述方法还包括：

在所述注意力机制卷积网络层中插入EgeConv点云局部特征提取算法；