Update cityscapes.py (#987)

Fix the bug that when split is 'val' and mode is 'test', it returns a PIL.Image.Image object instead of NDArray.
dmlc · Oct 17, 2019 · a78965f · a78965f
1 parent 99a58dd
commit a78965f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gluoncv/data/cityscapes.py b/gluoncv/data/cityscapes.py
@@ -48,6 +48,7 @@ def _class_to_index(self, mask):
     def __getitem__(self, index):
         img = Image.open(self.images[index]).convert('RGB')
         if self.mode == 'test':
+            img = self._img_transform(img)
             if self.transform is not None:
                 img = self.transform(img)
             return img, os.path.basename(self.images[index])