minor fix

Author: hangzhaomit

demo_test.sh

@@ -28,4 +28,4 @@ python3 -u test.py \
   --cfg config/ade20k-resnet50dilated-ppm_deepsup.yaml \
   DIR $MODEL_PATH \
   TEST.result ./ \
-  TEST.suffix _epoch_20.pth
+  TEST.checkpoint epoch_20.pth

eval_multipro.py

@@ -109,8 +109,6 @@ def worker(cfg, gpu_id, start_idx, end_idx, result_queue):
         weights=cfg.MODEL.weights_decoder,
         use_softmax=True)
 
-    net_encoder.features[0][0].weight.data = net_encoder.features[0][0].weight.data[:, (2,1,0), :, :] * 255.
-
     crit = nn.NLLLoss(ignore_index=-1)
 
     segmentation_module = SegmentationModule(net_encoder, net_decoder, crit)

utils.py

@@ -108,7 +108,7 @@ def unique(ar, return_index=False, return_inverse=False, return_counts=False):
     return ret
 
 
-def colorEncode(labelmap, colors, mode='BGR'):
+def colorEncode(labelmap, colors, mode='RGB'):
     labelmap = labelmap.astype('int')
     labelmap_rgb = np.zeros((labelmap.shape[0], labelmap.shape[1], 3),
                             dtype=np.uint8)