5dacca3ea8
* add 4 retinal vessel segmentation configs of UNet * fix flip augmentation * add unet benchmark on 4 medical datasets * fix hrf bug
7 lines
246 B
Python
7 lines
246 B
Python
_base_ = [
|
|
'../_base_/models/unet_s5-d16.py', '../_base_/datasets/stare.py',
|
|
'../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
|
|
]
|
|
test_cfg = dict(crop_size=(128, 128), stride=(85, 85))
|
|
evaluation = dict(metric='mDice')
|