File size: 383 Bytes
412c852 |
1 2 3 4 5 6 7 8 9 10 |
_base_ = './fcn_hr18_4xb4-40k_pascal-context-59-480x480.py'
model = dict(
pretrained='open-mmlab://msra/hrnetv2_w18_small',
backbone=dict(
extra=dict(
stage1=dict(num_blocks=(2, )),
stage2=dict(num_blocks=(2, 2)),
stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
|