snnetv2-semantic-segmentation / configs /pidnet /pidnet-l_2xb6-120k_1024x1024-cityscapes.py
HubHop
update
412c852
raw
history blame
434 Bytes
_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))