fix bugs

chenjun2hao · Jun 10, 2021 · bc0e193 · bc0e193
1 parent 79e13e5
commit bc0e193
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/experiments/cityscapes/ddrnet23.yaml b/experiments/cityscapes/ddrnet23.yaml
@@ -53,4 +53,4 @@ TEST:
   FLIP_TEST: true
   MULTI_SCALE: false
   MODEL_FILE: "pretrained_models/best_val.pth"
-  OUTPUT_INDEX: 0
+  OUTPUT_INDEX: 1
diff --git a/experiments/cityscapes/ddrnet23_slim.yaml b/experiments/cityscapes/ddrnet23_slim.yaml
@@ -55,4 +55,4 @@ TEST:
   SCALE_LIST: [1]
   #0.5,0.75,1.0,1.25,1.5,1.75
   # MODEL_FILE: "pretrained_models/best_val_smaller.pth"
-  OUTPUT_INDEX: 0
+  OUTPUT_INDEX: 1
diff --git a/experiments/cityscapes/ddrnet39.yaml b/experiments/cityscapes/ddrnet39.yaml
@@ -52,4 +52,4 @@ TEST:
   FLIP_TEST: true
   MULTI_SCALE: false
   MODEL_FILE: 
-  OUTPUT_INDEX: 0
+  OUTPUT_INDEX: 1
diff --git a/lib/models/ddrnet_23.py b/lib/models/ddrnet_23.py
@@ -337,7 +337,7 @@ def forward(self, x):
 
         if self.augment: 
             x_extra = self.seghead_extra(temp)
-            return [x_, x_extra]
+            return [x_extra, x_]
         else:
             return x_      
 

diff --git a/lib/models/ddrnet_23_slim.py b/lib/models/ddrnet_23_slim.py
@@ -337,7 +337,7 @@ def forward(self, x):
 
         if self.augment: 
             x_extra = self.seghead_extra(temp)
-            return [x_, x_extra]
+            return [x_extra, x_]
         else:
             return x_      
 

diff --git a/lib/models/ddrnet_39.py b/lib/models/ddrnet_39.py
@@ -357,7 +357,7 @@ def forward(self, x):
 
         if self.augment: 
             x_extra = self.seghead_extra(temp)
-            return [x_, x_extra]
+            return [x_extra, x_]
         else:
             return x_