Skip to content

Latest commit

 

History

History
39 lines (32 loc) · 2.12 KB

README.md

File metadata and controls

39 lines (32 loc) · 2.12 KB

Model weights and Visulaizations

You can download the learned weights and their results in the following links.

Dataset Visulization Weights
BRaTS 2018 [Download] [Download]
LA (fold 0) [Download] [Download]
Pancreas (fold 0) [Download] [Download]

BRaTS 2018 dataset

Download the dataset: Download

Training

python src/train_test.py -c brats-2018_lhunet_sgd

Before running the code above please adjust the config's paths.

Inference

Make sure in the config file you set the test_mode variable to true and give the path to the model you want to test in the ckpt_path.

python src/train_test.py -c brats-2018_lhunet_sgd

Pancreas and LA dataset

Download the pre-processed datasets with their splits: Pancreas, LA

Training

python src/train_test.py -c {dataset}_lhunet_sgd -f {fold_number}

instead of {dataset} you should write pancreas or la according to which dataset you want to train. regarding the {fold_number} for the Pancreas dataset it ranges from 0 till 3 (four-fold cross validation) and for the LA dataset it ranges from 0 till 4 (five-fold cross validation) Before running the code above please adjust the config's paths.

Inference

Make sure in the config file you set the just_test variable to true and give the path to the model you want to test in the ckpt_path.

python src/train_test.py -c {dataset}_lhunet_sgd -f {fold_number}