Skip to main content

Deep Learning tool for brain tumor segmentation.

Project description

Ensemble-of-Deep-2D-and-3D-Fully-Convolutional-Neural-Network-for-Brain-Tumor-Segmentation

Build Status PyPI version License: MIT

This repo utilize a ensemble of 2-D and 3-D fully convoultional neural network (CNN) for segmentation of the brain tumor and its constituents from multi modal Magnetic Resonance Images (MRI). The dense connectivity pattern used in the segmentation network enables effective reuse of features with lesser number of network parameters. On the BraTS validation data, the segmentation network achieved a whole tumor, tumor core and active tumor dice of 0.89, 0.76, 0.76 respectively.

Pipeline

pipeline

Results

Results

For training code please refer this repo

python version: 3.5

installation: pip install DeepBrainSeg

<hline>

Basic usage

for data in BraTs format

from DeepBrainSeg import deepSeg
segmentor = deepSeg(quick=True)
segmentor.get_segmentation_brats(path)
<hline>

for other formats

from DeepBrainSeg import deepSeg
t1_path = 
t2_path = 
t1ce_path = 
flair_path = 

segmentor = deepSeg(quick=True)
segmentor.get_segmentation(t1_path, t2_path, t1ce_path, flair_path, save = True)
<hline>

Steps followed for inference:

  • Our algorithm makes use of ANTs framework for mask generation. First call deepSeg class build ANTs framework locally in /tmp/DeepBrainSeg

  • Final segmentation is the result of ensemble of 4 different models:

    • ABLNet (modelABL.py, Air brain Lesion Network)

    • 3DBrainNet (model3DBNET.py, 3D multiresolution CNN)

    • Tiramisu2D (modelTis2D.py, 57 layered 2D CNN)

    • Tiramisu 3D (modelTir3D.py, 57 layered 3D CNN)

  • Extensive documentation will be uploaded soon, along with transfer learning framework

  • More details about network architecture and training procedure can be found here

Citation

If you use some of our work, please cite our work:

@inproceedings{kori2018ensemble,
  title={Ensemble of Fully Convolutional Neural Network for Brain Tumor Segmentation from Magnetic Resonance Images},
  author={Kori, Avinash and Soni, Mehul and Pranjal, B and Khened, Mahendra and Alex, Varghese and Krishnamurthi, Ganapathy},
  booktitle={International MICCAI Brainlesion Workshop},
  pages={485--496},
  year={2018},
  organization={Springer}
}

Contact

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for DeepBrainSeg, version 0.1.8
Filename, size File type Python version Upload date Hashes
Filename, size DeepBrainSeg-0.1.8-py3-none-any.whl (35.3 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size DeepBrainSeg-0.1.8.tar.gz (22.0 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page