arXiv:2507.05427cs.CV2025-07NeurIPS被引 13

让SAM2支持任意语言描述的图像分割,无需额外训练就能识别新类别。

OpenWorldSAM: Extending SAM2 for Universal Image Segmentation with Language Prompts

  • 用轻量视觉语言模型提取多模态嵌入,实现语言到空间的精准映射。
  • 仅训练450万参数,在COCO-stuff上达到顶尖零样本分割性能。
  • 支持多种语言提示,适合需要快速适配新类别的实际应用。

基于开放语言提示的物体分割仍是重大挑战,要求模型将文本语义精确映射到空间掩码,并处理多样且未见的类别。我们提出OpenWorldSAM,通过集成轻量级视觉语言模型(VLM)提取的多模态嵌入,扩展了提示驱动的Segment Anything Model v2(SAM2)至开放词汇场景。该方法遵循四大原则:统一提示,支持类别级与句子级语言描述;高效性,冻结SAM2和VLM预训练组件,仅在COCO-stuff数据集上训练450万参数;实例感知,引入新型位置消歧嵌入与交叉注意力层,提升多实例分割能力;泛化性,展现出强零样本能力,在未见类别与开放词汇概念上无需再训练即可良好表现。大量实验证明,OpenWorldSAM在多个基准上实现了开放词汇语义、实例与全景分割的最先进性能。代码已开源:https://github.com/GinnyXiao/OpenWorldSAM。

原文摘要 · Abstract (English)

The ability to segment objects based on open-ended language prompts remains a critical challenge, requiring models to ground textual semantics into precise spatial masks while handling diverse and unseen categories. We present OpenWorldSAM, a framework that extends the prompt-driven Segment Anything Model v2 (SAM2) to open-vocabulary scenarios by integrating multi-modal embeddings extracted from a lightweight vision-language model (VLM). Our approach is guided by four key principles: i) Unified prompting: OpenWorldSAM supports a diverse range of prompts, including category-level and sentence-level language descriptions, providing a flexible interface for various segmentation tasks. ii) Efficiency: By freezing the pre-trained components of SAM2 and the VLM, we train only 4.5 million parameters on the COCO-stuff dataset, achieving remarkable resource efficiency. iii) Instance Awareness: We enhance the model's spatial understanding through novel positional tie-breaker embeddings and cross-attention layers, enabling effective segmentation of multiple instances. iv) Generalization: OpenWorldSAM exhibits strong zero-shot capabilities, generalizing well on unseen categories and an open vocabulary of concepts without additional training. Extensive experiments demonstrate that OpenWorldSAM achieves state-of-the-art performance in open-vocabulary semantic, instance, and panoptic segmentation across multiple benchmarks. Code is available at https://github.com/GinnyXiao/OpenWorldSAM.

图像分割语言提示零样本SAM2

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