sct-pipeline / contrast-agnostic-softseg-spinalcord

Contrast-agnostic spinal cord segmentation project with softseg
MIT License
4 stars 3 forks source link

Augmentation with intensity scaling #69

Open naga-karthik opened 1 year ago

naga-karthik commented 1 year ago

Adding intensity-based scaling (for e.g. multiplying by -1 or non-linear scaling) to improve generalization on unseen contrasts (such as PSIR). Currently, we only use MONAI's RandIntensityScale transform but this is only applied with p=0.15. The range of this scaling [-0.25, 1] also needs to be a subject of discussion.

TODO: Revisit this for the model to be trained on the aggregated dataset