mmsegmentation/configs/unet/deeplabv3_unet_s5-d16_64x64_40k_drive.py
Junjun2016 49515e7e3c
Add more UNet-based medical segmentation benchmark (#324)
* Add UNet as backbone and FCN PSPNet DeepLabV3 as decode_head benchmark on 4 retinal vessel segmentation datasets

* adjust README of UNet
2021-01-11 00:07:59 -08:00

7 lines
254 B
Python

_base_ = [
'../_base_/models/deeplabv3_unet_s5-d16.py', '../_base_/datasets/drive.py',
'../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
]
test_cfg = dict(crop_size=(64, 64), stride=(42, 42))
evaluation = dict(metric='mDice')