中文

SurgPose:基于零样本学习与立体视觉的通用手术器械姿态估计

计算机视觉与模式识别 2025-05-19 v1 人工智能 机器学习 机器人学

摘要

在机器人辅助微创手术(RMIS)中,手术工具的精确姿态估计对于手术导航和机器人控制至关重要。传统的基于标记的方法虽然精度高,但面临遮挡、反射和工具特定设计的挑战。同样,监督学习方法需要大量标注数据集进行训练,限制了其对新工具的适应性。尽管零样本姿态估计模型在其他领域取得了成功,但在RMIS中用于手术器械姿态估计的研究仍属空白,这造成了在泛化到未见手术工具方面的差距。本文提出了一种新颖的手术器械6自由度(DoF)姿态估计流程,利用了最先进的零样本RGB-D模型,如FoundationPose和SAM-6D。我们通过采用RAFT-Stereo方法进行基于视觉的深度估计,提升了这些模型在反射和无纹理环境中的鲁棒深度估计能力。此外,我们通过用微调后的Mask R-CNN替换其实例分割模块Segment Anything Model(SAM),增强了SAM-6D,显著提高了在遮挡和复杂条件下的分割精度。大量验证表明,我们增强的SAM-6D在未见手术器械的零样本姿态估计方面超越了FoundationPose,为RMIS中的零样本RGB-D姿态估计设立了新的基准。这项工作增强了未见物体姿态估计的泛化能力,并开创了RGB-D零样本方法在RMIS中的应用。

关键词

引用

@article{arxiv.2505.11439,
  title  = {SurgPose: Generalisable Surgical Instrument Pose Estimation using Zero-Shot Learning and Stereo Vision},
  author = {Utsav Rai and Haozheng Xu and Stamatia Giannarou},
  journal= {arXiv preprint arXiv:2505.11439},
  year   = {2025}
}

备注

To be published in 2025 International Conference on Robotics and Automation (ICRA)