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