fix: roi size (Project-MONAI#75)

KumoLiu · Jul 1, 2022 · 2f881c6 · 2f881c6
1 parent 51a0a4f
commit 2f881c6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/SwinUNETR/BRATS21/utils/data_utils.py b/SwinUNETR/BRATS21/utils/data_utils.py
@@ -103,10 +103,10 @@ def get_loader(args):
             transforms.ConvertToMultiChannelBasedOnBratsClassesd(keys="label"),
             transforms.CropForegroundd(keys=["image", "label"], source_key="image", k_divisible=[args.roi_x,
                                                                                                  args.roi_y,
-                                                                                                 args.roi_x]),
+                                                                                                 args.roi_z]),
             transforms.RandSpatialCropd(keys=["image", "label"], roi_size=[args.roi_x,
                                                                            args.roi_y,
-                                                                           args.roi_x],
+                                                                           args.roi_z],
                                         random_size=False),
             transforms.RandFlipd(keys=["image", "label"], prob=0.5, spatial_axis=0),
             transforms.RandFlipd(keys=["image", "label"], prob=0.5, spatial_axis=1),

diff --git a/SwinUNETR/BTCV/utils/data_utils.py b/SwinUNETR/BTCV/utils/data_utils.py
@@ -85,7 +85,7 @@ def get_loader(args):
             transforms.RandCropByPosNegLabeld(
                 keys=["image", "label"],
                 label_key="label",
-                spatial_size=(args.roi_x, args.roi_y, args.roi_x),
+                spatial_size=(args.roi_x, args.roi_y, args.roi_z),
                 pos=1,
                 neg=1,
                 num_samples=4,

diff --git a/UNETR/BTCV/utils/data_utils.py b/UNETR/BTCV/utils/data_utils.py
@@ -84,7 +84,7 @@ def get_loader(args):
             transforms.RandCropByPosNegLabeld(
                 keys=["image", "label"],
                 label_key="label",
-                spatial_size=(args.roi_x, args.roi_y, args.roi_x),
+                spatial_size=(args.roi_x, args.roi_y, args.roi_z),
                 pos=1,
                 neg=1,
                 num_samples=4,