search

sct-pipeline / contrast-agnostic-softseg-spinalcord

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

Add mirroring test-time augmentation #96

Open naga-karthik opened 6 months ago

naga-karthik commented 6 months ago

This is a small PR that updates the inference script by adding mirroring test-time augmentation. Mirroing is also done in nnunet inference and @plbenveniste also noticed improved predictions for canproco SC seg preds.

naga-karthik commented 6 months ago

@plbenveniste do you think it's good to merge after the update of moving up torch.clamp before averaging?

plbenveniste commented 6 months ago

Looks good to me :)