arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2408.08870cs.CV

SAM2-UNet:Segment Anything 2 为自然和医学图像分割构建强大编码器

SAM2-UNet: Segment Anything 2 Makes Strong Encoder for Natural and Medical Image Segmentation

School of Computer Science and Engineering, Sun Yat-sen University(中山大学计算机科学与工程学院);School of Computer and Information Engineering, Jiangxi Normal University(江西师范大学计算机与信息工程学院);The Chinese University of Hong Kong (Shenzhen)(香港中文大学(深圳));Tianjin University(天津大学);Monash University(墨尔本大学);Pazhou Lab(琶洲实验室);Smart Hospital Research Institute, Peking University Shenzhen Hospital(北京大学深圳医院智能医院研究所)

Xinyu Xiong, Zihuang Wu, Shuangyi Tan, Wenxue Li, Feilong Tang, Ying Chen, Siying Li, Jie Ma, Guanbin Li

AI总结:

SAM2-UNet通过结合SAM2的Hiera主干和U型解码器,为自然和医学图像分割提供强大编码器,实现了高效且无需复杂调整的多任务分割性能。

摘要:

图像分割在视觉理解中扮演着重要角色。最近,新兴的视觉基础模型在各种任务上持续取得优异表现。在这一成功的基础上,本文证明了Segment Anything Model 2(SAM2)可以作为U型分割模型的强大编码器。我们提出了一种简单但有效的框架,称为SAM2-UNet,用于多用途图像分割。具体来说,SAM2-UNet采用SAM2的Hiera主干作为编码器,而解码器使用经典U型设计。此外,将适配器插入到编码器中,以允许参数高效的微调。在各种下游任务上的初步实验,如伪装物体检测、显著物体检测、海洋动物分割、镜像检测和息肉分割,表明我们的SAM2-UNet可以简单地超越现有的专门状态-of-the-art方法,无需额外的花哨功能。项目页面:\url{https://github.com/WZH0120/SAM2-UNet}。

英文摘要:

Image segmentation plays an important role in vision understanding. Recently, the emerging vision foundation models continuously achieved superior performance on various tasks. Following such success, in this paper, we prove that the Segment Anything Model 2 (SAM2) can be a strong encoder for U-shaped segmentation models. We propose a simple but effective framework, termed SAM2-UNet, for versatile image segmentation. Specifically, SAM2-UNet adopts the Hiera backbone of SAM2 as the encoder, while the decoder uses the classic U-shaped design. Additionally, adapters are inserted into the encoder to allow parameter-efficient fine-tuning. Preliminary experiments on various downstream tasks, such as camouflaged object detection, salient object detection, marine animal segmentation, mirror detection, and polyp segmentation, demonstrate that our SAM2-UNet can simply beat existing specialized state-of-the-art methods without bells and whistles. Project page: \url{https://github.com/WZH0120/SAM2-UNet}.

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

↑