Coder for "Enhancing Biventricular Segmentation with Multi-Scale UNet-Adapter and Consistency-Based Semi-Supervised Learning"
Please prepare an environment with python 3.9 torch 1.12.1+cu113 and torchaudio 0.12.1+cu113.
Please use the following code to install the required Python packages.
pip install -r requirements.txt
The dataset in our study can be found in ACDC. After downloading this dataset, run the script dataset/cdc_dataprocessing. py to process the data and obtain the ACDC_RLV dataset required for this study.
acdc_data_processing.py
Run the train script on ACDC_RLV dataset.
Train
train_model.py
Test
python test_model.py