[发明专利]多头注意力机制特征编码及双重判别解码的点云补全方法

说明书

本发明公开了多头注意力机制特征编码及双重判别解码的点云补全方法，其中针对局部特征感知的编码器模块，通过结合多头注意力机制和多层感知机对已有点云进行特征编码，自适应聚合输入点云的相似点特征，提高特征提取过程中对局部特征的感知；由判别器控制的双分支解码器，在局部特征及全局特征解码过程中设置判别器，能够基于输入点云特征分别对生成的骨架点云及细节点云进行判别，保证生成点云符合已有点云的全局特征及细节特征。本发明通过上述两模块共同作用可有效的增强点云形状补全效果，降低了生成点云的倒角距离损失，提高了生成点云的保真度。

技术领域

本发明涉及三维重建技术领域，特别涉及多头注意力机制特征编码及双重判别解码的点云补全方法。

背景技术

三维点云是三维场景和物体的直观表示，在各种视觉和机器人任务中有着广泛的应用。但是，由于遮挡和测量缺失等因素，扫描的3D点云通常是不完整的，这影响了实际应用，对不完整的点云进行补全成为了一项重要任务。因此，在硬件条件受限的情况下，基于深度学习的点云修补和补全工作是后续点云相关任务的关键与基础。

现有的基于深度学习的点云补全网络在点云补全中存在以下问题：

(1)对缺失区域的局部几何特征分析效果不好。

(2)使用注意力机制并未结合三维几何模型的数据特性。

以上问题的存在制约着三维点云模型的补全精度，不能满足对三维扫描仪获取的物体缺失的表面特征进行补全。

2021年，腾讯科技(深圳)有限公司的姚永强等人提出了一种点云数据处理方法、装置、计算机设备及存储介质(授权公告号：CN 113327279 B)，通过自相关性权重矩阵对点云坐标特征进行注意力编码，获取点云坐标特征对应的点位自相关性特征，预测待补全空间模型中的缺失部位获取缺失点部位特征，其缺点是，其方法对整体点云的几何特性进行编码，未对待补全模型的缺失区域的几何特征信息进行分析。

2022年，山东大学的郭亮等人提出了一种基于点云补全的点云配准方法及系统(授权公众号：CN 114004871 B)，通过利用注意机制融合源点云和目标点云的特征，使两个点云的语义信息相互补全，其优点是不需要对原始点云大量删减，且能够补全缺失的点云信息，实现高效、准确的配准。其缺点是，对现有点云的形状信息组合实现点云补全，不涉及对缺失部位局部几何特征的获取与未知形状信息的生成。

针对上述专利所存在的问题，2020年，清华大学的鲁继文等人提出了一种基于自注意力的多样点云补全方法和装置(申请公告号：CN112614071A)，通过将点云处理成为点代理序列并使用多头自注意力层构建点与点之间的关系，采用编码器解码器来构建点云不同点之间的长程关系实现点云重建，其缺点是通过点代理的方式会导致局部形状信息的感知效果不好。2022年，广东工业大学的高子淇等人提出了一种基于适应性自注意力变换网络的点云补全装置和方法(授权公告号：CN 113486988 B)，通过设计适应性自注意力变换模块对两次采样的空间信息特征进行适应性特征融合，进而对缺失点云进行补全，保证了轻量级的计算量与多重特征融合的有效性，其缺点是直接将用于自然语言处理的注意力变换模块用于点云处理，对于点云模型的补全效果不好。

综上所述，要提高三维点云待补全模型缺失区域的几何特征的表征、提取及生成补全效果，就必须对点云补全网络模型进行改进。

发明内容

本发明的目的是克服现有技术缺陷，提供多头注意力机制特征编码及双重判别解码的点云补全方法，能有效地提高三维待补全模型的缺失区域的补全精度。

本发明的目的是这样实现的：一种多头注意力机制特征编码及双重判别解码的点云补全方法，包括以下步骤：

步骤1)获取点云数据；

步骤2)基于多头注意力机制的局部特征编码；

步骤2-1)基于多层感知机获取点云特征；