Pre-training on large-scale T2w healthy/pathology data

ivadomed / model-seg-dcm

Segmentation of lesions on MRI scans in patients with Degenerative Cervical Myelopathy (DCM)

MIT License

3 stars 0 forks source link

Pre-training on large-scale T2w healthy/pathology data #11

Open naga-karthik opened 5 months ago

naga-karthik commented 5 months ago

This PR adds the code to do supervised (and self-supervised) pretraining on T2w MRI data from a collection of public/private, healthy/pathology datasets.

TODO:

[ ] Add code for self-supervised pre-training
[x] Test single-gpu and multi-gpu training