Skip to content

README.md

Latest commit

 

History

History
27 lines (19 loc) · 913 Bytes

README.md

File metadata and controls

27 lines (19 loc) · 913 Bytes

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):

# unet from scratch
mnms2_seg_train data.dir=~/.cache/cinema_datasets/mnms2/processed model.name=unet
# from scratch
mnms2_seg_train data.dir=~/.cache/cinema_datasets/mnms2/processed model.name=convunetr
# fine-tune
mnms2_seg_train data.dir=~/.cache/cinema_datasets/mnms2/processed model.name=convunetr model.ckpt_path=

Evaluation

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

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