mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #1 from smttsp/ISSUE-2337/bug/unet_param 

[FIX] fixing bug in `test_cfg` for U-Net _base_ script
pull/2347/head
Samet 2022-11-24 12:09:54 -05:00 committed by GitHub
parent 1bb58de01a 9387455d5b
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)))