15 lines
395 B
Python
15 lines
395 B
Python
_base_ = './segmenter_vit-s_mask_8xb1-160k_ade20k-512x512.py'
|
|
|
|
model = dict(
|
|
decode_head=dict(
|
|
_delete_=True,
|
|
type='FCNHead',
|
|
in_channels=384,
|
|
channels=384,
|
|
num_convs=0,
|
|
dropout_ratio=0.0,
|
|
concat_input=False,
|
|
num_classes=150,
|
|
loss_decode=dict(
|
|
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0)))
|