README.md

Multi-Disease, Multi-View & Multi-Center Right Ventricular Segmentation in Cardiac MRI (M&Ms2)

Preprocessing

Read the preprocessing readme on how to preprocess the data.

Training

The following command can be used to train/fine-tune the model on the dataset (please adjust the model.ckpt_path and data.dir accordingly):

# from scratch
mnms2_reg_train data.dir=~/.cache/cinema_datasets/mnms2/processed
# fine-tune
mnms2_reg_train data.dir=~/.cache/cinema_datasets/mnms2/processed model.ckpt_path=

Evaluation

The following command can be used to evaluate the model (please adjust the model.ckpt_path accordingly):

mnms2_reg_eval --data_dir ~/.cache/cinema_datasets/mnms2/processed --ckpt_path