Skip to content

Deep-learning based segmentation of the spinal cord lesion project

License

Notifications You must be signed in to change notification settings

ivadomed/model-seg-ms-axial-t2w

 
 

Repository files navigation

Assessing the Impact of Spinal Cord Curvature in Axial T2-weighted Intramedullary MS Lesion Segmentation

This repository contains the code for deep learning-based segmentation of the spinal cord and intramedullary MS lesions in Axial T2-weighted MRI scans. The model is based on the nnUNetv2 framework. This project is a collaboration between NeuroPoly (Polytechnique Montreal, Quebec) and TUM (Munich, Bavaria)

Model Overview

The model was trained on raw T2-weighted axial images of MS patients from multiple (four) sites. The TUM dataset is longitudinal (two sessions) and consisted of individual chunks (cervical, thoracic and lumbar) covering the entire spine. The three other sites used in this study were taken from the private sct-testing-large dataset from NeuroPoly. To ensure uniformity across sites, all images were initially re-oriented to RPI. Given an input image, the model is able to segment both the lesion and the spinal cord.

TODO: add a figure here

Using the model

Install dependencies

Once the dependencies are installed, download the latest model:

sct_deepseg -install-task seg_sc_lesion_t2w_ms

Getting the lesion and spinal cord segmentation

To segment a single image, run the following command:

sct_deepseg -i <INPUT> -task seg_sc_lesion_t2w_ms

For example:

sct_deepseg -i sub-001_T2w.nii.gz -task seg_sc_lesion_t2w_ms

The outputs will be saved in the same directory as the input image, with the suffix _lesion_seg.nii.gz for the lesion and _sc_seg.nii.gz for the spinal cord.

Analysis Pipeline

TODO:

Citation Info

If you find this work and/or code useful for your research, please cite our paper:

TODO:

About

Deep-learning based segmentation of the spinal cord lesion project

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 77.2%
  • Shell 22.8%