HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

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

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

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

Abstract

Image segmentation plays an important role in vision understanding. Recently,the emerging vision foundation models continuously achieved superiorperformance on various tasks. Following such success, in this paper, we provethat the Segment Anything Model 2 (SAM2) can be a strong encoder for U-shapedsegmentation models. We propose a simple but effective framework, termedSAM2-UNet, for versatile image segmentation. Specifically, SAM2-UNet adopts theHiera backbone of SAM2 as the encoder, while the decoder uses the classicU-shaped design. Additionally, adapters are inserted into the encoder to allowparameter-efficient fine-tuning. Preliminary experiments on various downstreamtasks, such as camouflaged object detection, salient object detection, marineanimal segmentation, mirror detection, and polyp segmentation, demonstrate thatour SAM2-UNet can simply beat existing specialized state-of-the-art methodswithout bells and whistles. Project page:\url{https://github.com/WZH0120/SAM2-UNet}.

Code Repositories

wzh0120/sam2-unet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-segmentation-on-mas3kSAM2-UNet
E-measure: 0.943
MAE: 0.021
S-measure: 0.903
mIoU: 0.799
image-segmentation-on-msd-mirror-segmentationSAM2-UNet
F-measure: 0.957
IoU: 0.918
MAE: 0.022
image-segmentation-on-pmdSAM2-UNet
F-measure: 0.826
IoU: 0.728
MAE: 0.027
image-segmentation-on-rmasSAM2-UNet
E-measure: 0.944
MAE: 0.022
S-measure: 0.874
mIoU: 0.738
salient-object-detection-on-dut-omron-2SAM2-UNet
E-measure: 0.912
MAE: 0.039
S-measure: 0.884
salient-object-detection-on-duts-te-1SAM2-UNet
E-measure: 0.959
MAE: 0.020
Smeasure: 0.934
salient-object-detection-on-ecssd-1SAM2-UNet
E-measure: 0.970
MAE: 0.020
S-measure: 0.950
salient-object-detection-on-hku-is-1SAM2-UNet
E-measure: 0.971
MAE: 0.019
S-measure: 0.941
salient-object-detection-on-pascal-s-1SAM2-UNet
E-measure: 0.931
MAE: 0.043
S-measure: 0.894

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp