MIC-DKFZ / nnUNet

Apache License 2.0
5.61k stars 1.7k forks source link

Final Contouring #2445

Open tedi14 opened 1 month ago

tedi14 commented 1 month ago

Hello,

I have been able to train a model on my own data. Now I want it to produce contours on an unseen CT data and want to be able to see it. Can anyone direct me to be able to do that? I am sorry if its a simple thing I am just not sure what to do next.

Thanks!

saikat-roy commented 1 month ago

Hey @tedi14. How about using np.gradient to calculate x, y and z gradients and adding them, of binarized output maps per class. This should give you contours of the segmented structures.

tedi14 commented 1 month ago

Hi @saikat-roy, I was hoping to run a few predictions on a few patient ct data sets and then take those predictions as nii.gz files and then convert them to dicom with their rstruct contours to examine them and see how the segmentations look. Is there a way to code this or run the first few steps getting a prediction segmentation of a new patient. What are the steps in running this?

Thanks!

LalithShiyam commented 1 month ago

Hi @tedi14,

Converting the nifti segmentations to an RT struct is not that straightforward, you can use PlatiPy, you can also try our tool, nifti2dicom, but we convert it to DICOM seg objects.

Cheers, Lalith