You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Nov 24, 2024. It is now read-only.
This repository is based on PyTorch 1.8.0, CUDA 11.1 and Python 3.8.10. All experiments in our paper were conducted on a single NVIDIA Tesla V100 GPU with an identical experimental setting.
Usage
Clone the repo.;
git clone https://github.com/ycwu1997/SS-Net.git
Put the data in './SS-Net/data';
Train the model;
cd SS-Net
# e.g., for 5% labels on LA
python ./code/train_ss_3d.py --labelnum 4 --gpu 0
Test the model;
cd SS-Net
# e.g., for 5% labels on LA
python ./code/test_LA.py --labelnum 4
Citation
If our SS-Net model is useful for your research, please consider citing:
@inproceedings{wu2022exploring,
title={Exploring Smoothness and Class-Separation for Semi-supervised Medical Image Segmentation},
author={Wu, Yicheng and Wu, Zhonghua and Wu, Qianyi and Ge, Zongyuan and Cai, Jianfei},
booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
pages={34--43},
volume={13435},
year={2022},
doi={10.1007/978-3-031-16443-9\_4},
organization={Springer, Cham}
}
Acknowledgements:
Our code is adapted from MC-Net, SemiSeg-Contrastive, VAT, and SSL4MIS. Thanks for these authors for their valuable works and hope our model can promote the relevant research as well.