Stop making 2D images 3D

fepegar · Dec 29, 2020 · ec19bc3 · ec19bc3
1 parent 2e7828d
commit ec19bc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/transforms/test_transforms.py b/tests/transforms/test_transforms.py
@@ -25,7 +25,7 @@ def get_transform(self, channels, is_3d=True, labels=True):
             tio.CropOrPad(cp_args),
             tio.ToCanonical(),
             tio.RandomAnisotropy(downsampling=(1.75, 2), axes=axes_downsample),
-            tio.EnsureShapeMultiple(2),
+            tio.EnsureShapeMultiple(2, method='crop'),
             tio.Resample((1, 1.1, 1.25)),
             tio.RandomFlip(axes=flip_axes, flip_probability=1),
             tio.RandomMotion(),