neuropoly / totalspineseg

Segmentation of vertebrae, intervertebral discs, spinal cord and CSF from MRI images.
5 stars 0 forks source link

MRI to CT model #17

Open plbenveniste opened 1 year ago

plbenveniste commented 1 year ago

The goal of this model is to accurately transform mri scans into ct scans. The greater objective is to be able to use Total Segmentator, which was trained on CT scans, on MRI images in order to perform segmentation.

To do:

plbenveniste commented 1 year ago

Added a new branch plb/mri_to_ct. Created a script to pre-process datasets for training and testing: preprocess_datasets.py. The goal is to use GoldAtlas and SynthRad2023 for training (because the first one has two modalities T1w and T2w and because the second one has a lot of data) and to validate/test on ICSFUS. The model chosen comes from this github : it is the denoising diffusion probabilistic model (DDPM) which proves to create great results when tranforming MRI to CT.

plbenveniste commented 10 months ago

Instead of the previous, I want to train a CycleGAN (Cycle-Consistent Generative Adversarial Network) or a UNIT (Unsupervised Image-to-Image Translation) model to see how these perform, before trying more complex models.