用视觉大模型实现室内机器人零样本物体实例分割
ZISVFM: Zero-Shot Object Instance Segmentation in Indoor Robotic Environments with Vision Foundation Models
- 结合SAM与自监督ViT,从深度图生成无对象先验的掩码提案
- 通过注意力特征过滤非物体掩码,再用K-Medoids聚类生成点提示
- 无需标注数据,在柜子、抽屉等复杂场景表现优异,适合真实机器人应用
服务机器人在非结构化环境中需有效识别并分割未知物体以提升功能。传统基于监督学习的分割方法依赖大量标注数据,难以应对现实场景中物体的多样性。未见物体实例分割(UOIS)方法通过合成数据训练模型以泛化至新物体,但常受仿真到现实的差距影响。本文提出ZISVFM框架,利用分割一切模型(SAM)的强大零样本能力及自监督视觉变换器(ViT)的显式视觉表示,分三阶段工作:(1) 使用SAM从着色深度图像生成无对象先验的掩码提案;(2) 利用自监督ViT的注意力特征过滤非物体掩码;(3) 采用K-Medoids聚类生成点提示,引导SAM实现精确物体分割。在两个基准数据集及自采集数据集上的实验验证了该方法在复杂环境(如橱柜、抽屉、手持物体)中的优越性能。源代码已开源。
原文摘要 · Abstract (English)
Service robots operating in unstructured environments must effectively recognize and segment unknown objects to enhance their functionality. Traditional supervised learningbased segmentation techniques require extensive annotated datasets, which are impractical for the diversity of objects encountered in real-world scenarios. Unseen Object Instance Segmentation (UOIS) methods aim to address this by training models on synthetic data to generalize to novel objects, but they often suffer from the simulation-to-reality gap. This paper proposes a novel approach (ZISVFM) for solving UOIS by leveraging the powerful zero-shot capability of the segment anything model (SAM) and explicit visual representations from a selfsupervised vision transformer (ViT). The proposed framework operates in three stages: (1) generating object-agnostic mask proposals from colorized depth images using SAM, (2) refining these proposals using attention-based features from the selfsupervised ViT to filter non-object masks, and (3) applying K-Medoids clustering to generate point prompts that guide SAM towards precise object segmentation. Experimental validation on two benchmark datasets and a self-collected dataset demonstrates the superior performance of ZISVFM in complex environments, including hierarchical settings such as cabinets, drawers, and handheld objects. Our source code is available at https://github.com/Yinmlmaoliang/zisvfm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。