Unverified Commit 8616227c authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix redundant transform in Cityscapes (#1045)

parent 5e42e805
......@@ -99,8 +99,6 @@ class Cityscapes(VisionDataset):
def __init__(self, root, split='train', mode='fine', target_type='instance',
transform=None, target_transform=None, transforms=None):
super(Cityscapes, self).__init__(root, transforms, transform, target_transform)
self.transform = transform
self.target_transform = target_transform
self.mode = 'gtFine' if mode == 'fine' else 'gtCoarse'
self.images_dir = os.path.join(self.root, 'leftImg8bit', split)
self.targets_dir = os.path.join(self.root, self.mode, split)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment