Save the label mask in the loop

pyplati · Dec 14, 2023 · 2890943 · 2890943
1 parent f3d6a99
commit 2890943
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/services/nnunet/service.py b/services/nnunet/service.py
@@ -159,10 +159,10 @@ def nnunet_service(data_objects, working_dir, settings):
 
                 sitk.WriteImage(mask, str(mask_file))
 
-            output_data_object = DataObject(
-                type="FILE", path=str(mask_file), parent=data_object
-            )
-            output_objects.append(output_data_object)
+                output_data_object = DataObject(
+                    type="FILE", path=str(mask_file), parent=data_object
+                )
+                output_objects.append(output_data_object)
 
         os.remove(io_path)