arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.13051cs.CVeess.IV

PA-SAM: 用于高质量图像分割的提示适配器SAM

PA-SAM: Prompt Adapter SAM for High-Quality Image Segmentation

Shanghai Jiao Tong University(上海交通大学);OPPO Research Institute(OPPO研究院);Innopeak Technology Inc(英诺达科技有限公司);The Hong Kong Polytechnic University(香港理工大学)

Zhaozhi Xie, Bochen Guan, Weihao Jiang, Muyang Yi, Yue Ding, Hongtao Lu, Lei Zhang

AI总结:

针对SAM在真实场景下掩码预测质量不足的问题,提出仅训练提示适配器的PA-SAM,通过双层面优化解码器特征提升分割性能,在多类分割任务上优于其他SAM类方法并已开源。

摘要:

Segment Anything Model(SAM,分割一切模型)在各类图像分割任务中展现出了优异性能。尽管SAM是在超过十亿个掩码的数据集上训练的,但它在诸多场景,尤其是真实场景下的掩码预测质量仍存在问题。本文提出一种新型提示驱动适配器,即Prompt Adapter Segment Anything Model(PA-SAM,提示适配器分割一切模型),旨在提升原始SAM的分割掩码质量。PA-SAM仅训练提示适配器,即可从图像中提取细节信息,并在稀疏与稠密提示两个层面优化掩码解码器特征,进而提升SAM的分割性能,生成高质量掩码。实验结果表明,本文的PA-SAM在高质量分割、零样本分割及开放集分割任务上均优于其他基于SAM的方法。我们已在https://github.com/xzz2/pa-sam 开源了源代码与模型。

英文摘要:

The Segment Anything Model (SAM) has exhibited outstanding performance in various image segmentation tasks. Despite being trained with over a billion masks, SAM faces challenges in mask prediction quality in numerous scenarios, especially in real-world contexts. In this paper, we introduce a novel prompt-driven adapter into SAM, namely Prompt Adapter Segment Anything Model (PA-SAM), aiming to enhance the segmentation mask quality of the original SAM. By exclusively training the prompt adapter, PA-SAM extracts detailed information from images and optimizes the mask decoder feature at both sparse and dense prompt levels, improving the segmentation performance of SAM to produce high-quality masks. Experimental results demonstrate that our PA-SAM outperforms other SAM-based methods in high-quality, zero-shot, and open-set segmentation. We're making the source code and models available at https://github.com/xzz2/pa-sam.

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

↑