Effective Semi-Supervised Medical Image Segmentation with Probabilistic Representations and Prototype Learning
by Yuchen Yuan, Xi Wang, Xikai Yang, and Pheng-Ann Heng
Official code for "Effective Semi-Supervised Medical Image Segmentation with Probabilistic Representations and Prototype Learning".
This repository is based on PyTorch 1.13.0, CUDA 11.7 and Python 3.7.13. All experiments in our paper were conducted on NVIDIA GeForce RTX 4090 GPU with an identical experimental setting.
We provide code
, data_split
for LA and ACDC dataset.
Data could be got at LA and ACDC.
To train a model,
python ./code/LA_train.py #for LA training
python ./code/ACDC_train.py #for ACDC training
To test a model,
python ./code/test_LA.py #for LA testing
python ./code/test_ACDC.py #for ACDC testing
Our code is largely based on our prior work SSCI and the work PRCL.
If you have any questions, welcome contact me at '[email protected]'