2020-09-04 15:35:52 +08:00
|
|
|
_base_ = '../deeplabv3plus/deeplabv3plus_r101-d8_512x512_160k_ade20k.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),
|
2022-10-18 20:13:53 +08:00
|
|
|
out_indices=(1, 2, 4, 6),
|
|
|
|
|
norm_cfg=dict(type='SyncBN', requires_grad=True)),
|
2020-09-04 15:35:52 +08:00
|
|
|
decode_head=dict(in_channels=320, c1_in_channels=24),
|
|
|
|
|
auxiliary_head=dict(in_channels=96))
|
