Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Fantastic Animals and Where to Find Them: Segment Any Marine Animal with Dual SAM

About

As an important pillar of underwater intelligence, Marine Animal Segmentation (MAS) involves segmenting animals within marine environments. Previous methods don't excel in extracting long-range contextual features and overlook the connectivity between discrete pixels. Recently, Segment Anything Model (SAM) offers a universal framework for general segmentation tasks. Unfortunately, trained with natural images, SAM does not obtain the prior knowledge from marine images. In addition, the single-position prompt of SAM is very insufficient for prior guidance. To address these issues, we propose a novel feature learning framework, named Dual-SAM for high-performance MAS. To this end, we first introduce a dual structure with SAM's paradigm to enhance feature learning of marine images. Then, we propose a Multi-level Coupled Prompt (MCP) strategy to instruct comprehensive underwater prior information, and enhance the multi-level features of SAM's encoder with adapters. Subsequently, we design a Dilated Fusion Attention Module (DFAM) to progressively integrate multi-level features from SAM's encoder. Finally, instead of directly predicting the masks of marine animals, we propose a Criss-Cross Connectivity Prediction (C$^3$P) paradigm to capture the inter-connectivity between discrete pixels. With dual decoders, it generates pseudo-labels and achieves mutual supervision for complementary feature representations, resulting in considerable improvements over previous techniques. Extensive experiments verify that our proposed method achieves state-of-the-art performances on five widely-used MAS datasets. The code is available at https://github.com/Drchip61/Dual_SAM.

Pingping Zhang, Tianyu Yan, Yang Liu, Huchuan Lu• 2024

Related benchmarks

TaskDatasetResultRank
Salient Object DetectionDUTS (test)
M (MAE)0.025
302
Camouflaged Object DetectionCOD10K (test)--
174
Marine Animal SegmentationMAS3K (test)
mIoU0.789
47
Marine Animal SegmentationRMAS (test)
mIoU73.5
47
Salient Object DetectionUSOD10k
S-alpha0.9238
40
Marine Animal SegmentationUFO120 (test)
mIoU81
35
Marine Animal SegmentationRUWI (test)
mIoU90.4
35
Underwater Salient Object DetectionUSOD10k 1.0 (test)
S_alpha0.9238
21
Medical Image SegmentationKvasir (test)
F1 Score90.9
18
Marine Animal SegmentationMAS3K
mIoU0.789
15
Showing 10 of 14 rows

Other info

Code

Follow for update