change start_level to 0

open-mmlab · Sep 30, 2021 · c8d1757 · c8d1757
1 parent 30ec997
commit c8d1757
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/_base_/models/fastfcn_r50-d32_jpu_psp.py b/configs/_base_/models/fastfcn_r50-d32_jpu_psp.py
@@ -18,7 +18,7 @@
         type='JPU',
         in_channels=(512, 1024, 2048),
         mid_channels=512,
-        start_level=1,
+        start_level=0,
         end_level=-1,
         dilations=(1, 2, 4, 8),
         align_corners=False,

diff --git a/mmseg/models/necks/jpu.py b/mmseg/models/necks/jpu.py
@@ -69,7 +69,7 @@ def __init__(self,
 
         self.conv_layers = nn.ModuleList()
         self.dilation_layers = nn.ModuleList()
-        for i in range(len(in_channels)):
+        for i in range(self.start_level, len(in_channels)):
             conv_layer = nn.Sequential(
                 ConvModule(
                     self.in_channels[i],