To develop a Convolutional Neural Network (CNN) segmentation model to diagnose brain tumor using Magnetic Resonance Imaging (MRI) images.
The project was run on Google Colab that provided a single 12GB NVIDIA Tesla K80 GPU.
These are 155 slices per MRI sequence of a particular patient.
- Numpy
- Matplotlib
- Keras
- SimpleITK
For downloading the dataset
- Go to "https://www.smir.ch/BRATS/Start2015".
- Register in there with official E-mail id.
- After confirming they will send the logins to your E-mail id.
- Login in there and go to "Challenges/BRATS2015".
- Then download the training and train dataset. But they won't provide the ground truth for test dataset.
In the codes folder following files are there :
- utils.py : It is the utility script containing the data loading and data augmentation code.
- simple_model.py : It is the utility script containing simple CNN model implemented in the paper.
- unet_model.py : It is the utility script containing the Unet model.
- Brain_segmentation.ipynb : It is notebook containing both the models executed 5 and 35 epochs respectively.
- Assets : It contains the sample dataset for just one patient and some files linked to notebook
- Rahul Kumar
© Rahul Kumar 2020
Licensed under the MIT License