mmsegmentation/configs/hrnet/fcn_hr18s_4xb4-160k_ade20k-512x512.py
谢昕辰 a3a144a361
[Refactor] Update config names (#1964)
* rename ann configs

* update ann yml

* update

* update

* update

* update

* update

* update ann readme

* update

* update deeplabv3

* update readme

* fix yml

* fix beit
2022-08-26 18:48:56 +08:00

10 lines
373 B
Python

_base_ = './fcn_hr18_4xb4-160k_ade20k-512x512.py'
model = dict(
pretrained='open-mmlab://msra/hrnetv2_w18_small',
backbone=dict(
extra=dict(
stage1=dict(num_blocks=(2, )),
stage2=dict(num_blocks=(2, 2)),
stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))