Artur112/Brain_Tumour_Segmentation

Code for training a 3DUnet for Brain tumour segmentation from Brats 2019 dataset; for feature extraction from the segmented volumes and for survival prediction. Run train.py for training, segment.py for segmenting test scans and evaluate.py for evaluating the performance of those segmentations. Basic code also written to perform survival prediction with a random forest classifiier.

Python