7 lines
241 B
Python
7 lines
241 B
Python
|
_base_ = './upernet_r50_4xb2-40k_cityscapes-512x1024.py'
|
||
|
|
model = dict(
|
||
|
|
pretrained='open-mmlab://resnet18_v1c',
|
||
|
|
backbone=dict(depth=18),
|
||
|
|
decode_head=dict(in_channels=[64, 128, 256, 512]),
|
||
|
|
auxiliary_head=dict(in_channels=256))
|