diff --git a/configs/_base_/models/deeplabv3_unet_s5-d16.py b/configs/_base_/models/deeplabv3_unet_s5-d16.py index 0cd262999..db4707cf4 100644 --- a/configs/_base_/models/deeplabv3_unet_s5-d16.py +++ b/configs/_base_/models/deeplabv3_unet_s5-d16.py @@ -47,4 +47,4 @@ model = dict( type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)), # model training and testing settings train_cfg=dict(), - test_cfg=dict(mode='slide', crop_size=256, stride=170)) + test_cfg=dict(mode='slide', crop_size=(256, 256), stride=(170, 170))) diff --git a/configs/_base_/models/fcn_unet_s5-d16.py b/configs/_base_/models/fcn_unet_s5-d16.py index a33e79728..0fb966c6c 100644 --- a/configs/_base_/models/fcn_unet_s5-d16.py +++ b/configs/_base_/models/fcn_unet_s5-d16.py @@ -48,4 +48,4 @@ model = dict( type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)), # model training and testing settings train_cfg=dict(), - test_cfg=dict(mode='slide', crop_size=256, stride=170)) + test_cfg=dict(mode='slide', crop_size=(256, 256), stride=(170, 170))) diff --git a/configs/_base_/models/pspnet_unet_s5-d16.py b/configs/_base_/models/pspnet_unet_s5-d16.py index fcff9ec4f..768f0fea0 100644 --- a/configs/_base_/models/pspnet_unet_s5-d16.py +++ b/configs/_base_/models/pspnet_unet_s5-d16.py @@ -47,4 +47,4 @@ model = dict( type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)), # model training and testing settings train_cfg=dict(), - test_cfg=dict(mode='slide', crop_size=256, stride=170)) + test_cfg=dict(mode='slide', crop_size=(256, 256), stride=(170, 170)))