This repository is the Pytorch implementation of "Multi-consistency for Semi-Supervised medical image segmentation with Diffusion Models"
We implemented our experiment on the super-parallel computer system of Guangxi University. The specific configuration is as follows:
- Centos 7.4
- NVIDIA Tesla V100 32G
- Intel Xeon gold 6230 2.1G 20C processor
Please modify your dataset path.
- ACDC: https://www.creatis.insa-lyon.fr/Challenge/acdc/databases.html | https://github.com/HiLab-git/SSL4MIS/tree/master/data/ACDC
- M&Ms: https://www.ub.edu/mnms
- Clone the repo:
git clone git@github.com:yunzhuC/MCSD.git
cd MCSD
- Train the model
python train.py
- Test the model
python test.py
Part of the code is revised from the UniMatch.
- The repository is being updated.