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 #1 from smttsp/ISSUE-2337/bug/unet_param

Browse Source 
[FIX] fixing bug in `test_cfg` for U-Net _base_ script
This commit is contained in:
Samet 2022-11-24 12:09:54 -05:00 committed by GitHub
commit bd87c92571
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)))