用粗略标注让SAM模型实现语义分割,大幅降低标注成本
Do Instance Priors Help Weakly Supervised Semantic Segmentation?
- 将类别掩码分解为连通域,沿物体骨架采样点提示
- 通过弱标签覆盖选择掩码,迭代优化伪标签提升精度
- 适配多种粗略标注,适合资源有限的语义分割场景
语义分割需要密集的像素级标注,获取成本高。为解决此问题,我们提出SeSAM框架,利用基础分割模型Segment Anything Model(SAM)与弱标签(包括粗略掩码、涂抹和点)结合。SAM原设计用于实例分割,无法直接用于语义分割。本文识别出SAM在该任务中的具体挑战,并确定适配其用于基于类别的分割的关键组件:将类别掩码分解为连通分量,沿物体骨架采样点提示,依据弱标签覆盖度选择SAM掩码,并通过伪标签迭代优化。集成半监督学习框架后,SeSAM平衡真实标签、SAM生成的伪标签及高置信度伪标签,显著提升分割质量。在多个基准和弱标注类型上实验表明,SeSAM持续优于弱监督基线,同时相比精细标注大幅降低标注成本。
原文摘要 · Abstract (English)
Semantic segmentation requires dense pixel-level annotations, which are costly and time-consuming to acquire. To address this, we present SeSAM, a framework that uses a foundational segmentation model, i.e. Segment Anything Model (SAM), with weak labels, including coarse masks, scribbles, and points. SAM, originally designed for instance-based segmentation, cannot be directly used for semantic segmentation tasks. In this work, we identify specific challenges faced by SAM and determine appropriate components to adapt it for class-based segmentation using weak labels. Specifically, SeSAM decomposes class masks into connected components, samples point prompts along object skeletons, selects SAM masks using weak-label coverage, and iteratively refines labels using pseudo-labels, enabling SAM-generated masks to be effectively used for semantic segmentation. Integrated with a semi-supervised learning framework, SeSAM balances ground-truth labels, SAM-based pseudo-labels, and high-confidence pseudo-labels, significantly improving segmentation quality. Extensive experiments across multiple benchmarks and weak annotation types show that SeSAM consistently outperforms weakly supervised baselines while substantially reducing annotation cost relative to fine supervision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。