arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.02420cs.CV
收藏

将SAM改造用于用户自定义语义感知分割

Repurposing SAM for User-Defined Semantics Aware Segmentation

University of California, Riverside(加州大学河滨分校)

Rohit Kundu, Sudipta Paul, Arindam Dutta, Amit K. Roy-Chowdhury

AI总结:

针对SAM缺乏语义感知、无法关联掩码与类别问题,提出U-SAM框架,利用合成或爬取图像学习掩码嵌入与类别映射,仅需类别名即可实现像素级语义分割,在两个基准上mIoU显著优于现有方法。

摘要:

Segment Anything Model(SAM,分割一切模型)擅长根据输入提示生成精确的目标掩码,但缺乏语义感知能力,无法将生成的掩码与特定目标类别关联。为解决这一局限,我们提出U-SAM——一种为SAM注入语义感知能力的新型框架,使其能为用户指定的目标类别生成针对性掩码。仅需用户提供目标类别名称作为输入,U-SAM即可为图像提供像素级语义标注,无需测试数据分布中的任何标注或未标注样本。我们的方法利用合成生成或网络爬取的图像积累所需目标类别的语义信息,随后学习SAM掩码嵌入与目标类别标签之间的映射函数,有效为SAM增强了粒度特定的语义识别能力。由此,用户可获得其请求的特定目标的有意义、针对性分割掩码,而非通用的无标注掩码。我们在PASCAL VOC 2012和MSCOCO-80上评估U-SAM,相比现有最优方法,其mIoU分别显著提升+17.95%和+5.20%。通过将SAM转化为语义感知分割模型,U-SAM为资源受限环境下多样且未见领域的像素级标注提供了实用灵活的解决方案。

英文摘要:

The Segment Anything Model (SAM) excels at generating precise object masks from input prompts but lacks semantic awareness, failing to associate its generated masks with specific object categories. To address this limitation, we propose U-SAM, a novel framework that imbibes semantic awareness into SAM, enabling it to generate targeted masks for user-specified object categories. Given only object class names as input from the user, U-SAM provides pixel-level semantic annotations for images without requiring any labeled/unlabeled samples from the test data distribution. Our approach leverages synthetically generated or web crawled images to accumulate semantic information about the desired object classes. We then learn a mapping function between SAM's mask embeddings and object class labels, effectively enhancing SAM with granularity-specific semantic recognition capabilities. As a result, users can obtain meaningful and targeted segmentation masks for specific objects they request, rather than generic and unlabeled masks. We evaluate U-SAM on PASCAL VOC 2012 and MSCOCO-80, achieving significant mIoU improvements of +17.95% and +5.20%, respectively, over state-of-the-art methods. By transforming SAM into a semantically aware segmentation model, U-SAM offers a practical and flexible solution for pixel-level annotation across diverse and unseen domains in a resource-constrained environment.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑