Add conv_seg init

open-mmlab · Jun 24, 2021 · 866cd69 · 866cd69
1 parent b24a981
commit 866cd69
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mmseg/models/decode_heads/setr_up_head.py b/mmseg/models/decode_heads/setr_up_head.py
@@ -28,8 +28,13 @@ def __init__(self,
                  num_convs=1,
                  up_scale=4,
                  kernel_size=3,
-                 init_cfg=dict(
-                     type='Constant', val=1.0, bias=0, layer='LayerNorm'),
+                 init_cfg=[
+                     dict(type='Constant', val=1.0, bias=0, layer='LayerNorm'),
+                     dict(
+                         type='Normal',
+                         std=0.01,
+                         override=dict(name='conv_seg'))
+                 ],
                  **kwargs):
 
         assert kernel_size in [1, 3], 'kernel_size must be 1 or 3.'