sct-pipeline / contrast-agnostic-softseg-spinalcord

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

Optimize DA #45

Closed jcohenadad closed 11 months ago

jcohenadad commented 1 year ago

https://github.com/sct-pipeline/contrast-agnostic-softseg-spinalcord/blob/1976cec36162f8dd1ce200cee13a3bf5493cbd9c/config_templates/meanGT_soft.json#L118

Ideas: artifacts, eg:

naga-karthik commented 1 year ago

We're sticking with the following data augmentation methods:

Training Screen Shot 2023-08-22 at 9 15 15 AM
Validation/Testing Screen Shot 2023-08-22 at 9 15 55 AM

I have noticed that adding more augmentations only increases the time per epoch with incremental improvements in performance.

EDIT: The transforms here are not up-to-date, please refer to the the next comment below.

naga-karthik commented 11 months ago

Here are the data augmentation transforms for the final model:

Training ![Screen Shot 2023-10-05 at 11 33 39 AM](https://github.com/sct-pipeline/contrast-agnostic-softseg-spinalcord/assets/53445351/f69862b4-bd8f-423a-bcdf-8f8ee0bffa60)
Validation/Testing ![Screen Shot 2023-10-05 at 11 35 02 AM](https://github.com/sct-pipeline/contrast-agnostic-softseg-spinalcord/assets/53445351/7ea91147-723d-4ec8-9a1d-367c4d348042)