Evidence Deep Learning and Dual Branch

This code is interoperable with WSL4MIS and ScribbleVC

Datasets

ACDC

1. The ACDC dataset with mask annotations can be downloaded from ACDC.
2. The scribble annotations of ACDC have been released in ACDC scribbles.
3. The pre-processed ACDC data used for training could be directly downloaded from ACDC_dataset.

MSCMR

1. The MSCMR dataset with mask annotations can be downloaded from MSCMRseg.
2. The scribble annotations of MSCMRseg have been released in MSCMR_scribbles.
3. The scribble-annotated MSCMR dataset used for training could be directly downloaded from MSCMR_dataset.

The slice classfication files have been available.

Requirements

Some important required packages include: Python 3.7 CUDA 11.1 Pytorch 1.13.1 torchvision 0.14.1 Some basic python packages such as Numpy, Scikit-image, SimpleITK, Scipy.

Training

To train the model, run this command:

python train_ACDC.py 

test

python test.py