tianrun-chen/SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts

What it solves

SAM-Adapter 解决了在挑战性的“性能下降场景”中使用 Segment Anything Model (SAM) 时出现的性能下降问题,例如伪装目标检测、阴影检测和医学图像分割(如息肉分割)。

How it works

它通过引入适配器机制,将预训练的 SAM 主干(支持 SAM、SAM2 和 SAM3)适配到下游任务。这使得模型在保持 SAM 通用分割能力的同时,能够在原模型可能失效的特定困难视觉环境中实现专门化。

Who it’s for

面向从事低对比度、伪装或医学影像等专门图像分割任务的计算机视觉研究者和开发者。

Highlights

  • 支持 SAM、SAM2 和 SAM3 主干。
  • 在伪装目标检测(COD10K、CAMO、CHAMELEON)和阴影检测(ISTD)上已验证有效性。
  • 可用于医学影像任务,如息肉分割(Kvasir 数据集)。
  • 提供不同 ViT 版本(ViT-H、ViT-L、ViT-B)的配置文件,以适配不同的 GPU 内存限制。