We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 99a58dd commit a78965fCopy full SHA for a78965f
gluoncv/data/cityscapes.py
@@ -48,6 +48,7 @@ def _class_to_index(self, mask):
48
def __getitem__(self, index):
49
img = Image.open(self.images[index]).convert('RGB')
50
if self.mode == 'test':
51
+ img = self._img_transform(img)
52
if self.transform is not None:
53
img = self.transform(img)
54
return img, os.path.basename(self.images[index])
0 commit comments