From 13d5a8d82c2ded62c3c9df8b8782711f53af60c0 Mon Sep 17 00:00:00 2001 From: Yiheng Wang Date: Thu, 15 Sep 2022 08:58:06 +0800 Subject: [PATCH 1/2] update ranzcr example according to 1.0 changes Signed-off-by: Yiheng Wang --- kaggle/RANZCR/4th_place_solution/models/seg_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kaggle/RANZCR/4th_place_solution/models/seg_model.py b/kaggle/RANZCR/4th_place_solution/models/seg_model.py index 34af59571d..c37b504022 100644 --- a/kaggle/RANZCR/4th_place_solution/models/seg_model.py +++ b/kaggle/RANZCR/4th_place_solution/models/seg_model.py @@ -69,7 +69,7 @@ def __init__( ): blocks.append( UpCat( - dim=dim, + spatial_dims=dim, in_chns=in_chn, cat_chns=skip_chn, out_chns=out_chn, From 7b83f23f3b98e8fa51dda8d42ef7d8c2bd45f40d Mon Sep 17 00:00:00 2001 From: Yiheng Wang Date: Sat, 8 Oct 2022 17:14:55 +0800 Subject: [PATCH 2/2] Fix type issue Signed-off-by: Yiheng Wang --- modules/dynunet_pipeline/transforms.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/modules/dynunet_pipeline/transforms.py b/modules/dynunet_pipeline/transforms.py index 9ef5ce07c7..52b95afcb3 100644 --- a/modules/dynunet_pipeline/transforms.py +++ b/modules/dynunet_pipeline/transforms.py @@ -67,11 +67,22 @@ def get_task_transforms(mode, task_id, pos_sample_num, neg_sample_num, num_sampl RandFlipd(["image", "label"], spatial_axis=[0], prob=0.5), RandFlipd(["image", "label"], spatial_axis=[1], prob=0.5), RandFlipd(["image", "label"], spatial_axis=[2], prob=0.5), + CastToTyped(keys=["image", "label"], dtype=(np.float32, np.uint8)), + EnsureTyped(keys=["image", "label"]), + ] + elif mode == "validation": + other_transforms = [ + CastToTyped(keys=["image", "label"], dtype=(np.float32, np.uint8)), + EnsureTyped(keys=["image", "label"]), ] - - return Compose(load_transforms + sample_transforms + other_transforms) else: - return Compose(load_transforms + sample_transforms) + other_transforms = [ + CastToTyped(keys=["image"], dtype=(np.float32)), + EnsureTyped(keys=["image"]), + ] + + all_transforms = load_transforms + sample_transforms + other_transforms + return Compose(all_transforms) def resample_image(image, shape, anisotrophy_flag): resized_channels = []