Remove deprecated img_size args in SwinUNETR (#416)

Remove deprecated `img_size` args in `SwinUNETR`
Add `allow_smaller=True` in CropForeground / CropForegroundd since
default value has been changed

Project-MONAI/MONAI#8430

---------

Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>

Author: KumoLiu

DAE/BTCV_Finetune/utils/data_utils.py

@@ -80,7 +80,7 @@ def get_loader(args):
             transforms.ScaleIntensityRanged(
                 keys=["image"], a_min=args.a_min, a_max=args.a_max, b_min=args.b_min, b_max=args.b_max, clip=True
             ),
-            transforms.CropForegroundd(keys=["image", "label"], source_key="image"),
+            transforms.CropForegroundd(keys=["image", "label"], source_key="image", allow_smaller=True),
             transforms.RandCropByPosNegLabeld(
                 keys=["image", "label"],
                 label_key="label",
@@ -111,7 +111,7 @@ def get_loader(args):
             transforms.ScaleIntensityRanged(
                 keys=["image"], a_min=args.a_min, a_max=args.a_max, b_min=args.b_min, b_max=args.b_max, clip=True
             ),
-            transforms.CropForegroundd(keys=["image", "label"], source_key="image"),
+            transforms.CropForegroundd(keys=["image", "label"], source_key="image", allow_smaller=True),
             transforms.ToTensord(keys=["image", "label"]),
         ]
     )

DAE/Pretrain_full_contrast/data/data_pretrain.py

@@ -21,7 +21,6 @@
     AsChannelFirstd,
     AsDiscrete,
     Compose,
-    CropForegroundd,
     LoadImaged,
     NormalizeIntensityd,
     Orientationd,

SwinMM/WORD/models/swin_unetr.py

@@ -38,7 +38,6 @@ def __init__(
 
         """
         super().__init__(
-            img_size,
             *args,
             num_heads=num_heads,
             feature_size=feature_size,

SwinUNETR/BRATS21/README.md

@@ -106,8 +106,7 @@ Mean Dice refers to average Dice of WT, ET and TC tumor semantic classes.
 A Swin UNETR network with standard hyper-parameters for brain tumor semantic segmentation (BraTS dataset) is be defined as:
 
 ``` bash
-model = SwinUNETR(img_size=(128,128,128),
-                  in_channels=4,
+model = SwinUNETR(in_channels=4,
                   out_channels=3,
                   feature_size=48,
                   use_checkpoint=True,

SwinUNETR/BRATS21/main.py

@@ -127,7 +127,6 @@ def main_worker(gpu, args):
     pretrained_pth = os.path.join(pretrained_dir, model_name)
 
     model = SwinUNETR(
-        img_size=(args.roi_x, args.roi_y, args.roi_z),
         in_channels=args.in_channels,
         out_channels=args.out_channels,
         feature_size=args.feature_size,

SwinUNETR/BRATS21/test.py

@@ -70,7 +70,6 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     pretrained_pth = os.path.join(pretrained_dir, model_name)
     model = SwinUNETR(
-        img_size=128,
         in_channels=args.in_channels,
         out_channels=args.out_channels,
         feature_size=args.feature_size,

SwinUNETR/BRATS21/utils/data_utils.py

@@ -99,7 +99,7 @@ def get_loader(args):
             transforms.LoadImaged(keys=["image", "label"]),
             transforms.ConvertToMultiChannelBasedOnBratsClassesd(keys="label"),
             transforms.CropForegroundd(
-                keys=["image", "label"], source_key="image", k_divisible=[args.roi_x, args.roi_y, args.roi_z]
+                keys=["image", "label"], source_key="image", k_divisible=[args.roi_x, args.roi_y, args.roi_z], allow_smaller=True
             ),
             transforms.RandSpatialCropd(
                 keys=["image", "label"], roi_size=[args.roi_x, args.roi_y, args.roi_z], random_size=False

SwinUNETR/BTCV/README.md

@@ -84,8 +84,7 @@ Once the json file is downloaded, please place it in the same folder as the data
 A Swin UNETR network with standard hyper-parameters for multi-organ semantic segmentation (BTCV dataset) is be defined as:
 
 ``` bash
-model = SwinUNETR(img_size=(96,96,96),
-                  in_channels=1,
+model = SwinUNETR(in_channels=1,
                   out_channels=14,
                   feature_size=48,
                   use_checkpoint=True,

SwinUNETR/BTCV/main.py

@@ -127,7 +127,6 @@ def main_worker(gpu, args):
 
     pretrained_dir = args.pretrained_dir
     model = SwinUNETR(
-        img_size=(args.roi_x, args.roi_y, args.roi_z),
         in_channels=args.in_channels,
         out_channels=args.out_channels,
         feature_size=args.feature_size,

SwinUNETR/BTCV/test.py

@@ -71,7 +71,6 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     pretrained_pth = os.path.join(pretrained_dir, model_name)
     model = SwinUNETR(
-        img_size=96,
         in_channels=args.in_channels,
         out_channels=args.out_channels,
         feature_size=args.feature_size,