Kelvinyu1117/glioma-segmentation

Brain Tumor Segmentation with BraTS 2018

Glioma Segmentation

EE4211 Computer Vision Project - Brain Tumor Segmentation with BraTS 2018

Data preprocessing

1. Download and extract the dataset under the subdirectory data

2. Run the pre-preprocessing.py

Train and Prediction

1. Upload all the generated pre-processed data to google drive in task1-full/task1-full, your directory structure should be the following:

   task1-full
   |---resnet_like_unet
   |---task1-full
   |---unet_batchNorm
   |---unet_original

2. Run the following colab notebooks:

   • model1-original unet

   • model2-batchnorm unet

   • model3-resnet-like unet

Acknowledgements and Citation

This project is developed with reference from the repo - Brain-tumor-segmentation-using-deep-learning authored by polo8214.