Skip to content

tbuikr/3D-SkipDenseSeg

Repository files navigation

3D-SkipDenseSeg

Skip-connected 3D DenseNet for volumetric infant brain MRI segmentation

By Toan Duc Bui, Jitae Shin, Taesup Moon

This is the implementation of our method in the MICCAI Grand Challenge on 6-month infant brain MRI segmentation-in conjunction with MICCAI 2017 in Pytorch.

Introduction

6-month infant brain MRI segmentation aims to segment the brain into: White matter, Gray matter, and Cerebrospinal fluid. It is a difficult task due to larger overlapping between tissues, low contrast intensity. We treat the problem by using very deep 3D convolution neural network. Our result achieved the top performance in 6 performance metrics.

Citation

@article{bui2019skip,
  title={Skip-connected 3D DenseNet for volumetric infant brain MRI segmentation},
  author={Bui, Toan Duc and Shin, Jitae and Moon, Taesup},
  journal={Biomedical Signal Processing and Control},
  volume={54},
  pages={101613},
  year={2019},
  publisher={Elsevier}
}

Requirements:

  • Pytorch >=0.4, python 3.0, Ubuntu 14.04
  • TiTan X Pascal 12GB

Installation

  • Step 1: Download the source code
https://github.com/tbuikr/3D-SkipDenseSeg.git
cd 3D-SkipDenseSeg
  • Step 2: Download dataset at http://iseg2017.web.unc.edu/download/ and change the path of the dataset data_path and saved path target_path in file prepare_hdf5_cutedge.py
data_path = '/path/to/your/dataset/'
target_path = '/path/to/your/save/hdf5 folder/'
  • Step 3: Generate hdf5 dataset
python prepare_hdf5_cutedge.py
  • Step 4: Run training
python train_v2.py

Run evaluation result.

python val.py

We also provide pretrained model. Use the pretrained model, you should achieve the result as the table.

Dice Coefficient (DC) for 9th subject (9 subjects for training and 1 subject for validation)

Pretrained CSF GM WM Average
3D-SkipDenseSeg 20000_model_3d_denseseg_v1 94.96 91.78 91.24 92.66

Run on testing set

python test.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages