From 9387455d5be7ae667b73f0fa26a58ec75714b1aa Mon Sep 17 00:00:00 2001 From: smttsp Date: Thu, 24 Nov 2022 11:58:17 -0500 Subject: [PATCH] [FIX] fixing bug in `test_cfg` for U-Net _base_ script tuples were required as default input, but int was given --- configs/_base_/models/deeplabv3_unet_s5-d16.py | 2 +- configs/_base_/models/fcn_unet_s5-d16.py | 2 +- configs/_base_/models/pspnet_unet_s5-d16.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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)))