Home

Awesome

Brain segmentation

This is a source code for the deep learning segmentation used in the paper Association of genomic subtypes of lower-grade gliomas with shape features automatically extracted by a deep learning algorithm. It employs a U-Net like network for skull stripping and FLAIR abnormality segmentation. This repository contains a set of functions for data preprocessing (MatLab), training and inference (Python). Weights for trained models are provided and can be used for deep learning based skull stripping or fine-tuning on a different dataset. If you use our model or weights, please cite:

@article{buda2019association,
  title={Association of genomic subtypes of lower-grade gliomas with shape features automatically extracted by a deep learning algorithm},
  author={Buda, Mateusz and Saha, Ashirbani and Mazurowski, Maciej A},
  journal={Computers in Biology and Medicine},
  volume={109},
  year={2019},
  publisher={Elsevier},
  doi={10.1016/j.compbiomed.2019.05.002}
}

Developed by mateuszbuda.

The repository is divided into two folders. One for skull stripping and one for FLAIR abnormality segmentation. They are based on the same model architecture but can be used separately.

Prerequisites

sudo pip install -r requirements.txt

Results

Below we show qualitative results for the average and median case. Blue outline corresponds to ground truth and red to the final automatic segmentation output. Images show FLAIR modality after preprocessing and skull stripping.

Average CaseMedian Case
Average caseMedian case

The distribution of Dice similarity coefficient (DSC) for the whole dataset of 110 cases used in our study.

DSC distribution

The red vertical line corresponds to mean DSC (83.60%) and the green one to median DSC (87.33%).

Trained weights

To download trained weights use download_weights.sh script located in both skull stripping or flair segmentation folder. It downloads *.h5 file with weights corresponding to training log shown in each task specific folder and responsible for the results reported there.

U-Net architecture

The figure below shows a U-Net architecture implemented in this repository.

unet

Data

brain-mri-lgg

kaggle.com/mateuszbuda/lgg-mri-segmentation