1.一种基于注意力机制的点云补全方法,其特征在于包括两个阶段:1)训练阶段:首先获取点云补全算法所需的数据集,包含有若干个残缺和完整的点云对,将残缺点云送入编码器提取得到残缺点云的特征向量,然后将特征向量送入解码器中,生成完整的点云;对生成的点云进行质量评估,计算生成点云和真实点云之间的倒角距离,作为损失函数指导更新编码器和解码器中的神经网络参数;2)实施阶段:获取实际扫描的点云数据,将点云数据送入在训练阶段已经训练好的编码器和解码器中,通过提取残缺点云的特征向量,生成完整的点云;所述的编码器为生成残缺点云特征向量:在获取到残缺点云数据P in 后,将其送入编码器中,编码器的结构依次为位置编码模块、4个串联的自适应全局特征聚合模块和一个最大值池化模块,位置编码模块由两个串联的多层感知器MLP层组成,位置编码模块的第二个多层感知器MLP层输出送入自适应全局特征聚合模块AGFA,自适应全局特征聚合模块共有四个,组成串联结构,最后一个自适应全局特征聚合模块4的输出结果分为二路,一路作为交叉全局特征聚合模块1的一个输入,另一路经过最大值池化操作后得到残缺点云特征向量f,残缺点云特征向量f用公式表达如下:f=Max-pool(AGFA (4) (MLP(P in )));式中的Max-pool代表最大值池化操作,上标(4)代表迭代通过4个串联的AGFA模块;所述的自适应全局特征聚合模块AGFA的结构为,将位置编码模块的第二个多层感知器MLP的输出作为“值”矩阵和“键”矩阵,同时将位置编码模块的第二个多层感知器MLP的输出的下采样作为“查询”矩阵,将“值”矩阵、“键”矩阵和“查询”矩阵一起送入多头注意力模块1中,多头注意力模块1由Pytorch深度学习工具包实现;然后将多头注意力模块1的输出结果与“查询”矩阵组成跨越连接,跨越连接后的输出经过归一化,送入多层感知器MLP2,多层感知器MLP2输出与归一化的输出再次构成第二次跨越连接,将第二次跨越连接的输出结果取代上次位置编码模块的第二个多层感知器MLP的输出,重复这一过程3次后得到聚合特征矩阵;将聚合特征矩阵送入平均值池化和多层感知器,求得通道注意力权重矩阵,通道注意力权重矩阵与聚合特征矩阵相乘得到自适应全局特征聚合模块的输出,随后将自适应全局特征聚合模块的输出送入最大值池化层,经最大值池化后,得到残缺点云特征向量f;所述的解码器为生成完整点云;在得到残缺点云特征向量后,将其送入解码器,解码器的结构依次为粗略点云生成模块、点云精细化模块1和点云精细化模块2,解码器的输出为最终的完整点云P out ;所述的粗略点云生成模块中,将残缺点云特征向量依次通过多层感知器和反卷积层聚合几何特征,将反卷积层输出和自适应全局特征聚合模块4的输出一起送入交叉全局特征聚合模块1,其中的交叉全局特征聚合模块1的结构为,将反卷积层输出作为“值”矩阵,“键”矩阵和“查询”矩阵,送入多头注意力模块2中,头注意力模块2由Pytorch深度学习工具包实现;再将多头注意力模块2输出结果与“查询”矩阵组成跨越连接,跨越连接后的输出经过归一化操作,送入多层感知器MLP3,多层感知器MLP3输出与归一化输出再次构成第二次跨越连接,将第二次跨越连接的输出结果取代反卷积层输出,然后重复3次该过程得到聚合特征矩阵;将聚合特征矩阵送入平均值池化和多层感知器MLP4,求得通道注意力权重矩阵,通道注意力权重矩阵与聚合特征矩阵相乘得到交叉全局特征聚合模块的输出;将交叉全局特征聚合模块的输出结果送入多层感知器MLP5,多层感知器MLP5的输出与原始的残缺点云按照点数并联在一起,即两个点云融合成一个点云,再经下采样操作得到粗略点云P coarse ;交叉全局特征聚合模块1的输出还作为送往点云精细化模块的交叉全局特征聚合模块2做前置模块输入;所述的点云精细化模块,将得到的粗略点云和得到的残缺点云特征向量分别送入多层感知器,两个多层感知器的输出并联,并联后的输出和交叉全局特征聚合模块1输出一起送入交叉全局特征聚合模块2中,交叉全局特征聚合模块2与交叉全局特征聚合模块1的结构完全相同,其输出再与粗略点云的多层感知器输出结果并联后,送入多层感知器MLP6进行特征融合,多层感知器MLP6特征融合后的输出与粗略点云构成跨层连接,输出精细点云,点云精细化模块2的结构与点云精细化模块1结构完全相同。
2.如权利要求1所述的一种基于注意力机制的点云补全方法,其特征在于:所述的生成点云的质量评估,采用倒角距离来计算生成点云和真实点云之间的差异,为损失函数指导更新训练阶段编码器和解码器中的神经网络参数,直到达到最大训练轮次;损失函数的计算公式如下:L=d CD (P coarse ,GT coarse )+d CD (P mid ,GT mid )+d CD (P out ,GT out );其中P coarse 、P mid 和P out 分别对应生成的粗略点云和两个精细点云,GT coarse 、GT mid 和GT out 为数据集中真实点云分别按照与生成点云相同点数降采样的结果;倒角距离的计算公式如下:其中P和Q代表两个点云,p和q分别是P点云和Q点云中的点。