validating-cnn-mgmt
Requirements
pyyaml
tqdm
numpy
pandas
dicom2nifti
antspyx
sklearn
torch
torchvision
torchio
monai
How to reproduce
-
Download the BraTS dataset. You must be registered and logged in to Kaggle.
-
Extract the zip file to the desired path. The extracted path should be passed to the
--source_dir
option in the preprocessing step. -
Run the preprocessing step for both train and test datasets:
python preprocess.py --source_dir path_to_brats_data --split train
python preprocess.py --source_dir path_to_brats_data --split test
- Run the validation experiments:
python validate_sweep.py
- To customize the validation experiment sweep, modify
config.yaml
.