Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
File size: 408 Bytes
b13b124
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
_base_ = './fcn_hr18_512x1024_40k_cityscapes.py'
model = dict(
    pretrained='open-mmlab://msra/hrnetv2_w48',
    backbone=dict(
        extra=dict(
            stage2=dict(num_channels=(48, 96)),
            stage3=dict(num_channels=(48, 96, 192)),
            stage4=dict(num_channels=(48, 96, 192, 384)))),
    decode_head=dict(
        in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))