基于双重注意力机制特征提取的深度高斯混合模型颅骨配准方法
文献发布时间:2023-06-19 10:58:46
技术领域
本发明属于三维点云模型配准相关技术领域,涉及一种基于双重注意力机制特征提取的深度高斯混合模型颅骨配准方法。
背景技术
随着三维获取技术的快速发展,获取到的三维点云数据从几何角度更好地再现了真实物体的形状信息,目前在逆向工程、计算机视觉、无人驾驶等实际应用领域中受到了广泛推广。三维点云配准是后续复原关键步骤之一,目的在于通过一个最佳变换矩阵估计,将位于不同坐标系的点云统一变换到同一坐标系下。
由于获取到的点云具有无序、结构不规则特性,已有方法将其转换为规则的体素网格便于处理,但是,会丢失一些重要几何信息。深度学习近些年得到了人们的普遍关注,利用深度学习直接处理点云会保留原始点云的关键信息。
注意力机制常用于二维图像分割、分类等应用领域中。在二维图像分割中,通过引入注意力机制来分别捕捉不同维度上的视觉特征关联,它更关注于找到输入数据中显著的与当前输出相关的有用信息,从而提高输出质量。该方法在二维图像上应用很广泛,但对于三维模型点云配准前期特征提取阶段应用却很少有相关研究。
深度高斯混合模型配准将点云配准问题定义为求两个高斯混合模型概率分布的KL散度最小值问题。主要思想是提取特征点与高斯混合模型参数之间的对应关系矩阵,矩阵中的元素代表某点属于高斯混合模型成分的概率,概率越大表示两者关联性越大,从而得到匹配参数,根据匹配参数恢复最优变换。
发明内容
针对现有技术的缺点或不足,本发明的目的在于提供一种基于双重注意力机制特征提取的深度高斯混合模型颅骨配准方法,学习到图片中更精细的特征并且减少运行的计算量,提高图像匹配精确度,提升算法运行速度,使秦腔脸谱特征识别更快、更准、效果更好。
为实现上述目的,本发明所采用的技术方案是:
基于双重注意力机制特征提取的深度高斯混合模型颅骨配准方法,其特征在于,包括以下步骤:
步骤1,通过三维扫描仪获取三维点云颅骨模型;
步骤2,将三维点云模型处理为只含顶点信息的1700个点的颅骨模型;
步骤3,将点云模型输入到卷积神经网络PointNet中,编码与解码阶段采用残差块+双重注意力机制通过对输入点云赋予不同权重提取关键特征,减少模型存储与计算开销;
残差网络由多个残差块串联构成,直接将输入信息跳跃连接到输出,注意力机制沿通道与空间两个方向分解,组合成双重注意力机制;
通道注意力机制采用平均池化和最大池化融合部分特征信息,生成平均池化特征
通过式(1)将两特征传播到只有一个隐藏层的多层感知机生成通道特征映射M
式(1)中σ是sigmoid激活函数,W
空间注意力机制沿通道轴应用平均池化和最大池化操作,融合上层输出的通道映射,生成平均池化特征
利用标准卷积层卷积以上两个特征,通过式(2)利用sigmoid激活函数生成有效特征描述符M
式(2)中,σ是sigmoid激活函数,f
步骤4,计算特征与高斯混合模型参数之间的对应关系矩阵得到匹配参数,通过式(3),(4),(5)计算高斯混合模型参数,包括权重、均值、方差;
式(3)中,α
式(4)中,μ
式(5)中,
步骤5,通过式(6)从匹配的参数中恢复最优变换:
进一步,步骤2中所述的对每个三维颅骨点云模型处理为只含顶点信息的1700个点。
本发明的有益效果是:
(1)本发明所采用的配准方法,通过获取点与模型参数之间的数据关联关系进行点云配准,通过一个置换不变网络中加入双重注意力机制强化特征提取得到对应关联矩阵,再通过两个模型参数单元恢复两个高斯混合模型之间的最优位姿变换,有效的建立了点与模型参数之间的数据关联关系,进一步提高了点云配准精度。克服了现有局部配准方法若无良好的初始化导致遇到较大变换匹配失败问题;
(2)本发明所采用的配准方法,基于双重注意力机制特征提取的深度高斯混合模型颅骨配准方法,有效地建立了点到模型之间的数据关联关系以实现点云高效配准,克服了现有全局配准方法速度慢且效率低问题。
附图说明
图1为本发明的流程图;
图2为三维扫描得到的模型;
图3(a)为处理后只含顶点信息1700个点的三维颅骨模型源点云图;
图3(b)为处理后只含顶点信息1700个点的三维颅骨模型目标点云图;
图4为计算得到的关系矩阵;
图5为从匹配参数恢复的最优变换过程图;
图6(a)为两点云初始位姿图;
图6(b)为实验配准结果图;
具体实施方式
如下面结合具体实施例对本发明作进一步详细描述,但不作为对本发明的限定。
本发明提出一种通过获取点与模型参数之间的数据关联关系进行点云配准方法,通过一个置换不变网络中加入双重注意力机制强化特征提取得到对应关联矩阵,再通过两个模型参数单元恢复两个高斯混合模型之间的最优位姿变换,有效的建立了点与模型参数之间的数据关联关系,提高了点云模型配准效率。
实施例1
本实施例提供了一种基于双重注意力机制特征提取的深度高斯混合模型颅骨配准方法,如图1所示,包括以下步骤:
步骤1,通过三维扫描仪获取三维点云模型;
本实例采用手持式三维激光扫描仪Handyscan3D扫描物体,获取三维模型信息,如图2所示。
步骤2,将三维点云模型处理为只含顶点信息的1700个点的颅骨模型,如图3所示,其中图3(a)表示源点云,图3(b)表示目标点云。
步骤3,将点云模型输入到卷积神经网络PointNet中,编码与解码阶段采用残差块+双重注意力机制通过对输入点云赋予不同权重提取关键特征,减少模型存储与计算开销:
残差网络由多个残差块串联构成,直接将输入信息跳跃连接到输出,注意力机制沿通道与空间两个方向分解,组合成双重注意力机制;
通道注意力机制采用平均池化和最大池化融合部分特征信息,生成平均池化特征
通过式(1)将两特征传播到只有一个隐藏层的多层感知机生成通道特征映射M
式(1)中σ是sigmoid激化函数,W
空间注意力机制沿通道轴应用平均池化和最大池化操作,融合上层输出的通道映射,生成平均池化特征
利用标准卷积层卷积以上两个特征,通过式(2)利用sigmoid激活函数生成有效特征描述符M
式(2)中,σ是sigmoid激化函数,f
步骤4,计算特征与高斯混合模型参数之间的对应关系矩阵得到匹配参数,通过式(3),(4),(5)计算高斯混合模型参数,包括权重、均值、方差,如图4所示。
式(3)中,α
式(4)中,μ
式(5)中,
步骤5,通过式(6)从匹配的参数中恢复最优变换,如图5所示。
参照上述实施例对本发明进行了详细说明,所属领域的普通技术人员应当理解:依然可以对本发明的具体实施方式进行修改或者等同替换,而未脱离本发明精神和范围的任何修改或者等同替换,其均应涵盖在本权利要求范围当中。