首页|混合注意力机制的非刚性三维点云模型对应关系计算

混合注意力机制的非刚性三维点云模型对应关系计算

扫码查看
针对非刚性三维点云模型对应关系构建中后处理过程计算复杂、算法泛化能力较差的问题,提出一种采用混合注意力机制并以无监督学习的方式计算对应关系的算法。首先引入点对特征改进边缘卷积,使提取的特征蕴含点与点之间更多的相似位姿信息;然后构建以余弦相似度计算为核心的混合注意力优化模块,将模型间特征相似的部分编码为相似度矩阵;最后,直接利用相似度矩阵与坐标信息双向重建对应的模型,获取最终的对应关系结果。在SURREAL,SHREC'19,SMAL和TOSCA数据集上的定性和定量实验结果表明,所提算法与CorrNet3D算法相比,在利用原始模型与重建模型之间的欧几里得距离衡量对应关系误差时,平均误差在SHREC'19和TOSCA数据集上分别减少了 0。19和5。00,在容忍误差为10%时对应关系准确率分别提高了 9。26个百分点和20。41个百分点,且在不同数据集上具有良好的泛化能力。
Correspondence Calculation of Non-Rigid 3D Point Shapes by Mixed Attention
Aiming at the complicated post-processing and poor generalization ability of correspondence cal-culation of non-rigid 3D point cloud shapes,a method that employs a mixed attention mechanism and unsu-pervised learning to calculate correspondence is proposed in this paper.First,the point pair feature improves the EdgeConv so that the extracted features can contain more similar pose information between points.Then,a mixed attention similarity refinement module is constructed by calculating cosine similarity,and the similar parts of features between models encode as a similarity matrix.Finally,the corresponding model is directly reconstructed in both directions using the similarity matrix and the coordinate information to compute the final correspondence.The qualitative and quantitative experimental results on SURREAL,SHREC'19,SMAL,and TOSCA datasets show that the proposed algorithm outperforms the CorrNet3D algorithm.Specifically,the av-erage error in measuring correspondence error using the Euclidean distance between the original and recon-structed shapes is reduced by 0.19 and 5.00 on the SHREC'19 and TOSCA datasets,respectively.The corres-pondence accuracy is also improved by 9.26 percentage points and 20.41 percentage points when the tolerance error is 10%.Furthermore,the proposed algorithm exhibits good generalization ability across different datasets.

3D point cloud shapesshape correspondenceunsupervised learningmixed attentionsimilarity matrix

杨军、张思洋、吴衍

展开 >

兰州交通大学电子与信息工程学院 兰州 730070

兰州交通大学测绘与地理信息学院 兰州 730070

福建技术师范学院大数据与人工智能学院 福清 350300

三维点云模型 对应关系 无监督学习 混合注意力 相似度矩阵

2024

计算机辅助设计与图形学学报
中国计算机学会

计算机辅助设计与图形学学报

CSTPCD北大核心
影响因子:0.892
ISSN:1003-9775
年,卷(期):2024.36(10)