Co-Seg++: Mutual Prompt-Guided Collaborative Learning for Versatile Medical Segmentation

ArXiv Paper

Qing Xu¹ Yuxiang Luo² Wenting Duan³ Zhen Chen^4✉

¹University of Nottingham   ²Sichuan University   ³Univeristy of Lincoln   ⁴HKISI, CAS  

^✉ Corresponding Author.

---

📰News

• [2025.06.03] We have released the code for Co-Seg++ !

🛠Setup

git clone https://github.com/xq141839/Co-Seg-Plus.git
cd Co-Seg-Plus
conda create -f Co-Seg-Plus.yaml

Key requirements: Cuda 12.2+, PyTorch 2.4+, mamba-ssm 2.1.0+

📚Data Preparation

• PUMA: Challenge Link

The data structure is as follows.

Co-Seg-Plus
├── datasets
│   ├── image_1024
│     ├── training_set_metastatic_roi_001.png
|     ├── ...
|   ├── mask_sem_1024
│     ├── training_set_metastatic_roi_001_nuclei.npy
|     ├── ...
|   ├── mask_ins_1024
│     ├── training_set_metastatic_roi_001_tissue.npy
|     ├── ...
|   ├── data_split.json

The json structure is as follows.

{ 
 "train": ['training_set_metastatic_roi_061.png'],
 "valid": ['training_set_metastatic_roi_002.png'],
 "test":  ['training_set_metastatic_roi_009.png'] 
 }

🎪Quickstart

• Train the Co-Seg++ with the default settings:

python train.py --dataset data/$YOUR DATASET NAME$ --sam_pretrain pretrain/$SAM2 CHECKPOINT$

Acknowledgements

• SAM2
• Medical-SAM-Adapter
• MambaVision