Upload upernet_global_base.yaml

Replace .py config files with .yaml versions

uniformer_base/upernet_global_base.yaml

@@ -0,0 +1,151 @@
+crop_size: !!python/tuple
+- 512
+- 512
+data:
+  samples_per_gpu: 2
+  test:
+    ann_dir: annotations/validation
+    data_root: data/ade/ADEChallengeData2016
+    img_dir: images/validation
+    pipeline:
+    - type: LoadImageFromFile
+    - flip: false
+      img_scale: !!python/tuple
+      - 2048
+      - 512
+      transforms:
+      - keep_ratio: true
+        type: Resize
+      - type: RandomFlip
+      - mean:
+        - 123.675
+        - 116.28
+        - 103.53
+        std:
+        - 58.395
+        - 57.12
+        - 57.375
+        to_rgb: true
+        type: Normalize
+      - keys:
+        - img
+        type: ImageToTensor
+      - keys:
+        - img
+        type: Collect
+      type: MultiScaleFlipAug
+    type: ADE20KDataset
+  workers_per_gpu: 4
+data_root: data/ade/ADEChallengeData2016
+dataset_type: ADE20KDataset
+img_norm_cfg:
+  mean:
+  - 123.675
+  - 116.28
+  - 103.53
+  std:
+  - 58.395
+  - 57.12
+  - 57.375
+  to_rgb: true
+model:
+  auxiliary_head:
+    align_corners: false
+    channels: 256
+    concat_input: false
+    dropout_ratio: 0.1
+    in_channels: 320
+    in_index: 2
+    loss_decode:
+      loss_weight: 0.4
+      type: CrossEntropyLoss
+      use_sigmoid: false
+    norm_cfg:
+      requires_grad: true
+      type: SyncBN
+    num_classes: 150
+    num_convs: 1
+    type: FCNHead
+  backbone:
+    attn_drop_rate: 0.0
+    drop_path_rate: 0.4
+    drop_rate: 0.0
+    embed_dim:
+    - 64
+    - 128
+    - 320
+    - 512
+    head_dim: 64
+    hybrid: false
+    layers:
+    - 5
+    - 8
+    - 20
+    - 7
+    mlp_ratio: 4.0
+    qkv_bias: true
+    type: UniFormer
+    windows: false
+  decode_head:
+    align_corners: false
+    channels: 512
+    dropout_ratio: 0.1
+    in_channels:
+    - 64
+    - 128
+    - 320
+    - 512
+    in_index:
+    - 0
+    - 1
+    - 2
+    - 3
+    loss_decode:
+      loss_weight: 1.0
+      type: CrossEntropyLoss
+      use_sigmoid: false
+    norm_cfg:
+      requires_grad: true
+      type: SyncBN
+    num_classes: 150
+    pool_scales: !!python/tuple
+    - 1
+    - 2
+    - 3
+    - 6
+    type: UPerHead
+  pretrained: null
+  test_cfg:
+    mode: whole
+  train_cfg: {}
+  type: EncoderDecoder
+norm_cfg:
+  requires_grad: true
+  type: SyncBN
+test_pipeline:
+- type: LoadImageFromFile
+- flip: false
+  img_scale: !!python/tuple
+  - 2048
+  - 512
+  transforms:
+  - keep_ratio: true
+    type: Resize
+  - type: RandomFlip
+  - mean:
+    - 123.675
+    - 116.28
+    - 103.53
+    std:
+    - 58.395
+    - 57.12
+    - 57.375
+    to_rgb: true
+    type: Normalize
+  - keys:
+    - img
+    type: ImageToTensor
+  - keys:
+    - img
+    type: Collect
+  type: MultiScaleFlipAug