11 lines
434 B
Python
11 lines
434 B
Python
_base_ = './pidnet-s_2xb6-120k_1024x1024-cityscapes.py'
|
|
checkpoint_file = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/pidnet/pidnet-l_imagenet1k_20230306-67889109.pth' # noqa
|
|
model = dict(
|
|
backbone=dict(
|
|
channels=64,
|
|
ppm_channels=112,
|
|
num_stem_blocks=3,
|
|
num_branch_blocks=4,
|
|
init_cfg=dict(checkpoint=checkpoint_file)),
|
|
decode_head=dict(in_channels=256, channels=256))
|