mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

[FIX] fixing bug in test_cfg for U-Net _base_ script

Browse Source 
tuples were required as default input, but int was given
This commit is contained in:
smttsp 2022-11-24 11:58:17 -05:00
parent 1bb58de01a
commit 9387455d5b
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/_base_/models/deeplabv3_unet_s5-d16.py
View File

@ -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)))

2
configs/_base_/models/fcn_unet_s5-d16.py
View File

@ -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)))

2
configs/_base_/models/pspnet_unet_s5-d16.py
View File

@ -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)))