计算机应用》唯一官方网站››2022,第42卷››问题(7): 2184-2191.内政部:10.11772/j.issn.1001-9081.2021071319

多媒体计算与计算机仿真 • 上一篇   

基于残差注意力机制的点云配准算法

秦庭威1,2,赵鹏程1,2,秦品乐1,2(),曾建朝1,2,柴锐1,2,黄永琦1,2  

  1. 1山西省医学影像人工智能工程技术研究中心中北大学),太原 030051
    2中北大学 大数据学院,太原 030051
  • 收稿日期:2021-07-22 修回日期:2021-10-13 接受日期:2021-10-18 发布日期:2021-11-01 出版日期:2022-07-10
  • 通讯作者:秦品乐
  • 作者简介:秦庭威(1997—),男,陕西渭南人,硕士研究生,共因失效会员,主要研究方向:点云配准、机器学习
    赵鹏程(1995—),男,陕西渭南人,硕士,共因失效会员,主要研究方向:三维点云处理、计算机视觉
    曾建朝(1963—),男,山西太原人,教授,博士,共因失效会员,主要研究方向:复杂系统的维护决策和健康管理
    柴锐(1985—),男,山西运城人,讲师,博士,共因失效会员,主要研究方向:医学影像处理
    黄永琦(1997—),男,山西太原人,硕士研究生,共因失效会员,主要研究方向:点云配准、计算机视觉。
  • 基金资助:
    山西省重点研发计划项目(201803D31212-1);山西省工程技术研究中心建设项目(201805D121008)

基于剩余注意机制的点云配准算法

秦廷伟1,2,赵鹏程1,2,品乐琴1,2(),曾建超1,2,芮柴(Rui CHAI)1,2,黄永琪1,2  

  1. 1山西医学影像与数据分析工程研究中心(中国北方大学),太原山西030051
    2中国北方大学数据科学与技术学院,山西太原030051
  • 收到:2021-07-22 修订过的:2021-10-13 认可的:2021-10-18 在线:2021-11-01 出版:2022-07-10
  • 联系人:品乐琴
  • 关于作者:秦廷伟,1997年生,硕士研究生。他的研究兴趣包括点云注册、机器学习。
    赵鹏程,1995年生,硕士,主要研究方向为三维点云处理、计算机视觉。
    曾建超,1963年生,博士,教授。他的研究兴趣包括复杂系统的维护决策和健康管理。
    蔡瑞,1985年出生,博士,讲师。他的研究兴趣包括医学图像处理。
    黄永琪,1997年生,硕士研究生。他的研究兴趣包括点云配准、计算机视觉。
  • 支持单位:
    山西省重点研发计划(201803D31212-1);山西省工程技术研究中心建设项目(201805D121008)

摘要:

针对传统点云配准算法精度低、鲁棒性差以及放疗前后癌症患者无法实现精确放疗的问题,提出一种基于残差注意力机制的点云配准算法(ADGCNNLK)首先,在动态图深度卷积网络(DGCNN)中添加残差注意力机制来有效地利用点云的空间信息,并减少信息损失;然后,利用添加残差注意力机制的DGCNN(美国有线电视新闻网)提取点云特征,这样做不仅可以在保持点云置换不变性的同时捕捉点云的局部几何特征,也可以在语义上将信息聚合起来,从而提高配准效率;最后,将提取到的特征点映射到高维空间中并使用经典的图像迭代配准算法斯里兰卡进行配准。实验结果表明,所提算法与迭代最近点算法(ICP)、ICP(Go-ICP)和PointNetLK相比,在无噪、有噪的情况下配准效果均最好。其中,在无噪情况下,与点网络LK相比,所提算法的旋转均方误差降低了74.61%,平移均方误差降低了47.50%;在有噪声的情况下,与点网络LK相比,所提算法的旋转均方误差降低了73.13%,平移均方误差降低了44.18%,说明所提算法与点网络LK相比鲁棒性更强。将所提算法应用于放疗前后癌症患者人体点云模型的配准,从而辅助医生治疗,并实现了精确放疗。

关键词: 点云配准, 特征提取, 残差注意力机制, 深度学习, 放疗

摘要:

针对传统点云配准算法精度低、鲁棒性差以及癌症患者放疗前后无法进行精确放疗的问题,提出了一种注意力动态图卷积神经网络Lucas-Kanade(ADGCNNLK)。首先,在动态图卷积神经网络(DGCNN)中加入剩余注意机制,有效利用点云的空间信息,减少信息损失。然后,使用添加了剩余注意机制的DGCNN提取点云特征,该过程不仅能够捕获点云的局部几何特征,同时保持点云替换的不变性,而且能够从语义上聚合信息,从而提高了点云特征的提取效率注册效率。最后,将提取的特征点映射到高维空间,并使用经典的图像迭代配准算法LK(Lucas Kanade)对节点进行配准。实验结果表明,与迭代最近点(ICP)、全局最优ICP(Go-ICP)和PointNetLK相比,无论有无噪声,该算法都具有最佳的配准效果。其中,在无噪声的情况下,与PointNetLK相比,该算法的旋转均方误差降低了74.61%,平移均方误差减少了47.50%在有噪声的情况下,与PointNetLK相比,该算法的旋转均方误差降低了73.13%,平移均方误差降低了44.18%,表明该算法比PointNetLK更具鲁棒性。并将该算法应用于癌症患者放疗前后人体点云模型的配准,辅助医生进行治疗,实现精确放疗。

关键词: 点云注册, 特征提取, 剩余注意机制, 深度学习, 放射治疗

中图分类号: