contrast-agnostic-softseg-spinalcord v2.4

Improved version of contrast-agnostic spinal cord segmentation model trained on healthy subjects and pathologies in the cervical cord:

• Contrasts: T1w, T2w, T2star, MTon-MTS, MToff-MTS, DWI (averaged), mp2rage UNIT1, PSIR, STIR
• Pathologies: multiple sclerosis (MS) patients, compressed spinal cords in degenerative cervical myelopathy (DCM) patients.

Works well on:

• Spinal cord injury (SCI) lesions
• GRE-EPI images
• B0 Field Map images
• Lumbar cord
• PSIR and STIR contrasts

Main difference from version v2.3 is the addition of lumbar T2w images and PSIR/STIR contrasts of MS patients.

The train/val/test splits from all the datasets used to train this model can be found in the datasplits folder in the source code. Further details on the number of training samples across all datasets and samples per contrast can be found in dataset_splits.md.

EDIT: the initial .zip file containing the model was corrupted, hence a new (fixed) .zip was uploaded

What's Changed

• Update preprocessing script for spine-generic with new naming convention by @sandrinebedard in #105

Full Changelog: v2.3...v2.4

contrast-agnostic-softseg-spinalcord v2.3

Improved version of contrast-agnostic spinal cord segmentation model trained on healthy subjects and pathologies in the cervical cord:

• Contrasts: T1w, T2w, T2star, MTon-MTS, MToff-MTS, DWI (averaged), mp2rage UNIT1
• Pathologies: multiple sclerosis (MS) patients, compressed spinal cords in degenerative cervical myelopathy (DCM) patients.

Works well on:

• Spinal cord injury (SCI) lesions
• GRE-EPI images
• B0 Field Map images
• Lumbar cord

Main difference from earlier versions of contrast-agnostic is the addition of pathological data (MS, DCM) to the training set.

The train/val/test splits from all the datasets used to train this model can be found in the datasplits folder in the source code. Further details on the number of training samples across all datasets and samples per contrast can be found in dataset_stats.md.

Full Changelog: v2.2...v2.3

contrast-agnostic-softseg-spinalcord v2.3.1

Official code for the spin-off extension of contrast-agnostic spinal cord segmentation comparing different DL architectures including CNNs, Vision Transformers and ConvNeXT models accepted as a short paper at MIDL 2024. The paper can be found here.

contrast-agnostic-softseg-spinalcord v2.2

What's Changed

• Update preprocessing script for spine-generic with new naming convention by @sandrinebedard in #105

Full Changelog: v2.1...v2.2

contrast-agnostic-softseg-spinalcord v2.1

About

This release updates the training codebase and adds a newer variant of the contrast-agnostic model (details below).

What's Changed

• Update readme with citation info and arxiv badge by @naga-karthik in #88
• Fixed documentation on inference by @jcohenadad in #91
• Fixed documentation on inference (follow-up) by @jcohenadad in #93
• Make inference CLI callable from anywhere by @jcohenadad in #95
• add info about dataset split used for training contrast-agnostic model by @naga-karthik in #97
• Simplify inference script by @naga-karthik in #100
• Port towards config file-based training by @naga-karthik in #90

Other notable changes

• The model in this release is trained with binarized soft labels (hence the name soft_bin) as opposed to directly training on soft labels as in the model in release v2.0
• In addition to the monai-based nnunet model, this release also adds the feature to train other models as well (e.g. SwinUNETR, MedNeXT, etc.)
• Three new classes of CSA evaluation scripts are added -- (1) evaluating CSA across different models, (2) evaluating CSA across different resolutions, and (3) evaluating CSA across different resolutions.
  • A unified script analyze_csa_across.py is added for generating CSA violin plots across different classes mentioned above.

Full Changelog: v2.0...v2.1

contrast-agnostic-softseg-spinalcord v2.0

About

This release contains the official code for the submission to Medical Image Analysis Journal. The model weights are uploaded as release assets along with all the scripts for preprocessing, training, CSA and QC generation.

What's Changed

• Add subjects with missing contrasts to exclude by @sandrinebedard in #53
• Add script to compute CSA binarized and run QC on binarized prediction for ivadomed by @sandrinebedard in #58
• Modify post processing nnunet for CSA computation and QC report by @sandrinebedard in #57
• Naga/nnunet by @sandrinebedard in #65
• add monai-based scripts for dataset conversion, training, and inference by @naga-karthik in #60
• Clean up processing and repository by @sandrinebedard in #64
• add requirements for gpu inference by @naga-karthik in #76
• add qc generation script for epi data by @naga-karthik in #77
• Update main readme by @naga-karthik in #78
• Upgrade and pin monai to 1.3.0 by @naga-karthik in #79
• Finalise script for figures by @sandrinebedard in #80
• add anima metrics by @naga-karthik in #81
• Add anova test on soft all and hard all by @sandrinebedard in #82

New Contributors

• @naga-karthik made their first contribution in #60

Full Changelog: v1.2...v2.0

contrast-agnostic-softseg-spinalcord v1.2 - MICCAI 2023

This code was used for the submission to MICCAI 2023. QC reports of the tested models on Basel-MP2RAGE and sci-colorado datasets are includes as release assets.

What's Changed

• use _pred, not _pred_painted & remove 4th dimension by @sandrinebedard in #35
• CSA charts preliminary notebook by @ebadrian in #30
• Add command for CSA charts by @sandrinebedard in #43
• Change file names for MTS contrasts by @sandrinebedard in #42
• Fix MTS filenames in preprocessing and charts scripts by @sandrinebedard in #46
• Add processing of datasets with pathology by @sandrinebedard in #50
• Training and evaluation scripts MICCAI 2023 by @ebadrian in #51

New Contributors

• @ebadrian made their first contribution in #30

Full Changelog: v1.1...v1.2

contrast-agnostic-softseg-spinalcord v1.1

What's Changed

• The registration algorithm was modified for the soft segmentation generation in the dataset spine-generic/data-multi-subject.

contrast-agnostic-softseg-spinalcord v1.0

First release.

Used for the creation of soft segmentations on the internal database spine-generic-processed.