mmsegmentation/configs/unet/unet-s5-d16_fcn_4xb4-40k_ch...

10 lines
342 B
Python

_base_ = [
'../_base_/models/fcn_unet_s5-d16.py', '../_base_/datasets/chase_db1.py',
'../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
]
crop_size = (128, 128)
data_preprocessor = dict(size=crop_size)
model = dict(
data_preprocessor=data_preprocessor,
test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))