fix bugs

Author: chenjun2hao

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

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

experiments/cityscapes/ddrnet39.yaml

@@ -52,4 +52,4 @@ TEST:
   FLIP_TEST: true
   MULTI_SCALE: false
   MODEL_FILE: 
-  OUTPUT_INDEX: 0
+  OUTPUT_INDEX: 1

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_      
 

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_      
 

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_