fix #45

MIC-DKFZ · Aug 9, 2023 · ae16068 · ae16068
1 parent bfc50c1
commit ae16068
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HD_BET/data_loading.py b/HD_BET/data_loading.py
@@ -84,7 +84,7 @@ def save_segmentation_nifti(segmentation, dct, out_fname, order=1, dtype=np.uint
                      bbox[2][0]:bbox[2][1]] = segmentation
     else:
         seg_old_size = segmentation
-    if np.any(np.array(seg_old_size) != np.array(dct['size'])[[2, 1, 0]]):
+    if np.any([i != j for i, j in zip(np.array(seg_old_size), np.array(dct['size'])[[2, 1, 0]])]):
         seg_old_spacing = resize_segmentation(seg_old_size, np.array(dct['size'])[[2, 1, 0]], order=order)
     else:
         seg_old_spacing = seg_old_size