mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 06:36:08 +08:00
c11da07e18
* delete convert function and add instruction to README.md * unified model convert and README * remove url * fix import error * fix unittest * rename pretrain * rename vit and deit pretrain * Update upernet_deit-b16_512x512_160k_ade20k.py * Update upernet_deit-b16_512x512_80k_ade20k.py * Update upernet_deit-b16_ln_mln_512x512_160k_ade20k.py * Update upernet_deit-b16_mln_512x512_160k_ade20k.py * Update upernet_deit-s16_512x512_160k_ade20k.py * Update upernet_deit-s16_512x512_80k_ade20k.py * Update upernet_deit-s16_ln_mln_512x512_160k_ade20k.py * Update upernet_deit-s16_mln_512x512_160k_ade20k.py Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com> Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
9 lines
401 B
Python
9 lines
401 B
Python
_base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py'
|
|
|
|
model = dict(
|
|
pretrained='pretrain/deit_small_patch16_224-cd65a155.pth',
|
|
backbone=dict(num_heads=6, embed_dims=384, drop_path_rate=0.1),
|
|
decode_head=dict(num_classes=150, in_channels=[384, 384, 384, 384]),
|
|
neck=dict(in_channels=[384, 384, 384, 384], out_channels=384),
|
|
auxiliary_head=dict(num_classes=150, in_channels=384))
|