uniformer_image_segmentation / configs /mobilenet_v2 /deeplabv3_m-v2-d8_512x512_160k_ade20k.py
SakuraD's picture
update
bd22556
raw
history blame
No virus
408 Bytes
_base_ = '../deeplabv3/deeplabv3_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),
out_indices=(1, 2, 4, 6)),
decode_head=dict(in_channels=320),
auxiliary_head=dict(in_channels=96))