christianpayer / MedicalDataAugmentationTool-VerSe

GNU General Public License v3.0
177 stars 81 forks source link

Save Output Images for segmentation during inference #13

Open Alistair-dv opened 3 years ago

Alistair-dv commented 3 years ago

Hi!

I'm finding that the _seg.png image outputed in inference/main_vertebrae_segementation.py function 'test' is blank. The file size is 2KB, far smaller than the 50KB input image that is generated just before.

I've checked that the variable prediction_labels_resampled is correct and the _seg.nii.gz output is correct too.

Any ideas on how I could fix this?