mmsegmentation/configs/mobilenet_v2/mobilenet-v2-d8_fcn_4xb2-80...

13 lines
405 B
Python

_base_ = '../fcn/fcn_r101-d8_4xb2-80k_cityscapes-512x1024.py'
model = dict(
pretrained='mmcls://mobilenet_v2',
backbone=dict(
_delete_=True,
type='MobileNetV2',
widen_factor=1.,
strides=(1, 2, 2, 1, 1, 1, 1),
dilations=(1, 1, 1, 2, 2, 4, 4),
out_indices=(1, 2, 4, 6)),
decode_head=dict(in_channels=320),
auxiliary_head=dict(in_channels=96))