Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Region-based nnUNet training #7

Open
2 of 7 tasks
plbenveniste opened this issue Feb 22, 2024 · 1 comment
Open
2 of 7 tasks

Region-based nnUNet training #7

plbenveniste opened this issue Feb 22, 2024 · 1 comment
Assignees
Labels

Comments

@plbenveniste
Copy link
Collaborator

plbenveniste commented Feb 22, 2024

In this issue, I document my experiments in training and testing a region-based nnUNet.
The model takes into input a Nifti image and outputs the spinal cord (value=1) and the MS lesions (value=2).
The fact that it is region-based means that lesions must be in the spinal cord.

The necessary steps to take are :

  • Segment SC for images which have lesion seg but no SC seg (with SCT v6.2)
  • Manually correct the SC segs
  • Build an nnUNet dataset
  • Train a 2D region-based nnUNet (with normal loss and no smoothing loss)
  • Train a 3D region-based nnUNet (with normal loss and no smoothing loss)
  • Test model on test set
  • Compare model performances
@plbenveniste plbenveniste self-assigned this Feb 22, 2024
@plbenveniste plbenveniste changed the title Region-based nnUNet Region-based nnUNet training Feb 22, 2024
@plbenveniste
Copy link
Collaborator Author

In the CanProCo dataset as well as in the stc-testing-large dataset, some subjects were missing a SC segmentation.

  • For the 33 CanProCo subjects, the SC was segmented using the region-based model trained on CanProCo (release).
    They were then manually corrected and pushed to CanProCo on branch plb/add_sc_seg_for_regionbased
  • For the 186 sct-testing-large subjects, the SC was segmented using the contrast-agnostic-sc-seg model from SCT v6.2. They were then manually corrected and pushed to sct-testing-large on branch plb/add_sc_seg_for_regionbased

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant