Home

Awesome

SemiSAM

Official repository of BIBM'24 paper "SemiSAM: Enhancing Semi-Supervised Medical Image Segmentation via SAM-Assisted Consistency Regularization".

Introduction

:computer: Usage

  1. Clone the repo
git clone https://github.com/YichiZhang98/SemiSAM
cd SemiSAM
  1. Put the data in data/2018LA_Seg_Training Set and SAM checkpoint in ckpt/sam_med3d.pth.

  2. Train the model

cd code
python train_LA_semisam_mt.py
  1. Test the model
python test_LA.py

:books: Citation

If you find this paper useful, please consider citing:

@inproceedings{SemiSAM,
  title={SemiSAM: Enhancing Semi-Supervised Medical Image Segmentation via SAM-Assisted Consistency Regularization},
  author={Zhang, Yichi and Yang, Jin and Liu, Yuchen and Cheng, Yuan and Qi, Yuan},
  booktitle={2024 IEEE International Conference on Bioinformatics and Biomedicine (BIBM)},
  year={2024}
}