Minor bug fix in preprocessor.py

NVIDIA · Feb 26, 2024 · 34c2381 · 34c2381
1 parent 38934f9
commit 34c2381
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PyTorch/Segmentation/nnUNet/data_preprocessing/preprocessor.py b/PyTorch/Segmentation/nnUNet/data_preprocessing/preprocessor.py
@@ -217,7 +217,7 @@ def collect_spacings(self):
         spacing = np.array(spacing)
         target_spacing = np.median(spacing, axis=0)
         if max(target_spacing) / min(target_spacing) >= 3:
-            lowres_axis = np.argmin(target_spacing)
+            lowres_axis = np.argmax(target_spacing)
             target_spacing[lowres_axis] = np.percentile(spacing[:, lowres_axis], 10)
         self.target_spacing = list(target_spacing)