Git Product home page Git Product logo

brain-segmentation's Introduction

Project Overview

  • Dataset used: MRBrainS18
  • 148 is used as validation

About dataset

Classes label
Cortical gray matter 1
Basal ganglia 2
White matter 3
White matter lesions 4
Cerebrospinal fluid in the extracerebral space 5
Ventricles 6
Cerebellum 7
Brain stem 8

Preprocessing steps

     
  • Registered and Bias Field Correction was already done in the dataset
  • Skull stripping was done only for T1 weighted MRI using DeepBrain library which creates a mask for skull removal.
  • Furthermore, contrast of T1 weighted MRI was improved using Histogram Equalization technique
Regularized Biased Field Corrected MRI Removed Skull Histogram equalization

Approach

Cortical gray matter, White matter, Cerebrospinal fluid in the extracerebral space

Cortical gray matter, White matter, Cerebrospinal fluid in the extracerebral space can be easily reduced by appling thresholding to T1- weighted MRI further a small U-Net was used to denoise the threshold image.

Total parameters: 60,553  

Rest

For rest of the classes training was done on a custom model inspired by Unet Architecture. The model has 3 encoders stacked together in bottleneck layer and then a single decoder. There are skip connections from encoder to decoder to enhance segmentation.

Total parameters: 151,717

Key difference between U-Net and Architecture used

U-Net Architecture used
Only one encoder and one decoder Three encoder and one decoder
Deep architecture with about 10 Million parameters Shallow with about 600 Thousands parameters
Doesn’t have dilated convolution layers Has dilated convolution layers

Loss Function

Dice coefficient is used as Loss function in final training though Jaccard distance and crossentropy were also tried.

Learning Curves

Results (On Validation Data)


Label 1 2 3 4 5 6 7 8
Dice coefficient 0.702 0.758 0.770 0.746 0.704 0.882 0.887 0.855

References


This project was made as part of the Smart India hackathon 2018 - Software Edition, a 36 hour hackathon organised by Government of India. The problem statement was given by Department of Atomic Energy, India

brain-segmentation's People

Contributors

ubamba98 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.