Merge remote-tracking branch 'yotamyaniv/sitkcompressed'

MIC-DKFZ · Oct 6, 2023 · ef4e3da · ef4e3da
2 parents 8a8bfd0 + 82e825d
commit ef4e3da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnunetv2/imageio/simpleitk_reader_writer.py b/nnunetv2/imageio/simpleitk_reader_writer.py
@@ -126,4 +126,4 @@ def write_seg(self, seg: np.ndarray, output_fname: str, properties: dict) -> Non
         itk_image.SetOrigin(properties['sitk_stuff']['origin'])
         itk_image.SetDirection(properties['sitk_stuff']['direction'])
 
-        sitk.WriteImage(itk_image, output_fname)
+        sitk.WriteImage(itk_image, output_fname, True)