chenyzzz / MCSD

Multi-consistency for Semi-Supervised medical Image Segmentation with Diffusion Model

Multi-consistency for Semi-Supervised medical image segmentation with Diffusion Models

by Yunzhu Chen

Introduction

This repository is the Pytorch implementation of "Multi-consistency for Semi-Supervised medical image segmentation with Diffusion Models"

Requirements

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

Usage

1. Clone the repo:

git clone https://github.com/chenyzzz/MCSD.git
cd MCSD

2.. Train the model

python train.py

4. Test the model

python test.py

Citation

Acknowledgement

Part of the code is revised from the UniMatch.

Note

• The repository is being updated.