This is used for the first paper. It's been reimplemented in Pytorch and the original source is implemented in Tensorflow.
- The code has been written in Python 3.6.3 and Pytorch 1.0.1
- Make sure to install all the libraries given in requirement.txt, and you can do so by the following command
pip install -r requirement.txt
BRATS 2018 dataset was chosen to substantiate our proposed method. It contains the 3D multimodal brain MRI data of 285 labeled training subjects and 66 testing subjects which have to be submitted for online validation. We randomly split the 285 labeled training data into training and testing set with the rate of 4:1.
- Download the BRATS 2018 data and place it in data folder.(Visit this link to downlaod the data. You need to register for the challenge)
- To perform bias field correction, you have to perform correction for each directory of dataset.
$ cd data_preprocessing
$ python n4correction Brats2018/MICCAI_BraTS_2018_Data_Training/HGG
$ python train_valid_score.py --id=FullyHierarchical