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

Merge pull request #2347 from smttsp/master

Browse Source 
BUG FIX in test_cfg for U-Net _base_ scripts
This commit is contained in:
Miao Zheng 2022-11-28 12:07:26 +08:00 committed by GitHub
commit a2e9191016
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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)))