mmsegmentation/configs/upernet/upernet_r18_4xb2-40k_citysc...

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))