arXiv:2409.15481cs.ROcs.CV2024-09被引 6

用少量数据让SAM精准分割没见过的物体,尤其在遮挡和纹理复杂场景下表现优异。

Adapting Segment Anything Model for Unseen Object Instance Segmentation

  • 通过热图生成点提示,实现无需类别信息的精准前景定位
  • 仅用10%训练数据即达顶尖性能,显著降低标注成本
  • 适合需要快速适配新物体的机器人视觉任务

未见物体实例分割(UOIS)对在非结构化环境中运行的自主机器人至关重要。以往方法需在大规模桌面数据集上全监督预训练才能有效。本文提出UOIS-SAM,一种数据高效的UOIS解决方案,充分利用SAM的高精度与强泛化能力。该方法集成两个核心组件:(i) 基于热图的提示生成器(HPG),生成类无关的点提示并实现精确前景预测;(ii) 层次化判别网络(HDNet),适配SAM的掩码解码器,缓解基线模型中存在的背景混淆与过度分割问题,尤其在遮挡和纹理丰富物体场景中表现突出。在OCID、OSD以及包含PhoCAL和HouseCat6D等光照挑战性数据集上的大量实验表明,即使仅使用先前方法10%的训练样本,UOIS-SAM仍能达到未见物体分割的最先进水平,验证了其在多种桌面场景下的有效性与鲁棒性。

原文摘要 · Abstract (English)

Unseen Object Instance Segmentation (UOIS) is crucial for autonomous robots operating in unstructured environments. Previous approaches require full supervision on large-scale tabletop datasets for effective pretraining. In this paper, we propose UOIS-SAM, a data-efficient solution for the UOIS task that leverages SAM's high accuracy and strong generalization capabilities. UOIS-SAM integrates two key components: (i) a Heatmap-based Prompt Generator (HPG) to generate class-agnostic point prompts with precise foreground prediction, and (ii) a Hierarchical Discrimination Network (HDNet) that adapts SAM's mask decoder, mitigating issues introduced by the SAM baseline, such as background confusion and over-segmentation, especially in scenarios involving occlusion and texture-rich objects. Extensive experimental results on OCID, OSD, and additional photometrically challenging datasets including PhoCAL and HouseCat6D, demonstrate that, even using only 10% of the training samples compared to previous methods, UOIS-SAM achieves state-of-the-art performance in unseen object segmentation, highlighting its effectiveness and robustness in various tabletop scenes.

实例分割SAM少样本学习机器人视觉

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。