bugfix

liaw05 · Apr 12, 2019 · 82383d0 · 82383d0
1 parent 813967a
commit 82383d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/batchgenerators/augmentations/utils.py b/batchgenerators/augmentations/utils.py
@@ -690,7 +690,7 @@ def pad_nd_image(image, new_shape=None, mode="constant", kwargs=None, return_sli
     pad_above = difference // 2 + difference % 2
     pad_list = [[0, 0]]*num_axes_nopad + list([list(i) for i in zip(pad_below, pad_above)])
 
-    if (not all([[i == 0] for i in pad_below])) and (not all([[i == 0] for i in pad_above])):
+    if not ((all([i == 0 for i in pad_below])) and (all([i == 0 for i in pad_above]))):
         res = np.pad(image, pad_list, mode, **kwargs)
     else:
         res = image