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] |
Download the dataset: Download
python src/train_test.py -c brats-2018_lhunet_sgd
Before running the code above please adjust the config's paths.
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
Download the pre-processed datasets with their splits: Pancreas, LA
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.
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}