arXiv:2505.11980cs.CVcs.AI2025-05AAAI被引 15

让SAM自动找提示点,提速图像分割且不降精度

AoP-SAM: Automation of Prompts for Efficient Segmentation

  • 用轻量模型自动定位关键区域生成最优提示
  • 测试时采用粗到精采样过滤,提升效率并减少冗余计算
  • 无需微调即可保持零样本泛化能力,适合自动化场景

Segment Anything Model (SAM) 是强大的图像分割基础模型,通过提示工程实现鲁棒的零样本泛化。然而,依赖人工提示在真实场景中不切实际,尤其在需要快速提供提示和资源高效的应用中。本文提出 AoP-SAM,一种自动化生成 SAM 所需提示的新方法,可自动在最优位置生成关键提示。该方法采用轻量级提示预测模型,检测图像中的关键实体并识别放置提示候选的最佳区域,利用 SAM 的图像嵌入,保持其零样本泛化能力且无需微调。此外,引入测试时实例级自适应采样与过滤机制,以粗到精方式生成提示,显著提升提示与掩码生成效率,降低计算开销并减少冗余掩码优化。三个数据集上的评估表明,AoP-SAM 显著提升了提示生成效率与掩码生成精度,使 SAM 更适用于自动化分割任务。

原文摘要 · Abstract (English)

The Segment Anything Model (SAM) is a powerful foundation model for image segmentation, showing robust zero-shot generalization through prompt engineering. However, relying on manual prompts is impractical for real-world applications, particularly in scenarios where rapid prompt provision and resource efficiency are crucial. In this paper, we propose the Automation of Prompts for SAM (AoP-SAM), a novel approach that learns to generate essential prompts in optimal locations automatically. AoP-SAM enhances SAM's efficiency and usability by eliminating manual input, making it better suited for real-world tasks. Our approach employs a lightweight yet efficient Prompt Predictor model that detects key entities across images and identifies the optimal regions for placing prompt candidates. This method leverages SAM's image embeddings, preserving its zero-shot generalization capabilities without requiring fine-tuning. Additionally, we introduce a test-time instance-level Adaptive Sampling and Filtering mechanism that generates prompts in a coarse-to-fine manner. This notably enhances both prompt and mask generation efficiency by reducing computational overhead and minimizing redundant mask refinements. Evaluations of three datasets demonstrate that AoP-SAM substantially improves both prompt generation efficiency and mask generation accuracy, making SAM more effective for automated segmentation tasks.

图像分割自动化提示SAM效率优化

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