fix bug

lib/core/function.py

@@ -213,6 +213,7 @@ def test(config, test_dataset, testloader, model,
             image, size, name = batch
             size = size[0]
             pred = test_dataset.multi_scale_inference(
+                config,
                 model,
                 image,
                 scales=config.TEST.SCALE_LIST,

lib/datasets/cityscapes.py

@@ -192,7 +192,7 @@ class Cityscapes(BaseDataset):
 
     def save_pred(self, preds, sv_path, name):
         palette = self.get_palette(256)
-        preds = np.asarray(np.argmax(preds, axis=1), dtype=np.uint8)
+        preds = np.asarray(np.argmax(preds.cpu(), axis=1), dtype=np.uint8)
         for i in range(preds.shape[0]):
             pred = self.convert_label(preds[i], inverse=True)
             save_img = Image.fromarray(pred)