Skip to content

GitLab

"...en/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "ec1aded12e8fbdc98d6703f9111ed824addebc77"
Unverified Commit 8616227c authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix redundant transform in Cityscapes (#1045)

parent 5e42e805
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
torchvision/datasets/cityscapes.py
View file @ 8616227c
...@@ -99,8 +99,6 @@ class Cityscapes(VisionDataset): ...@@ -99,8 +99,6 @@ class Cityscapes(VisionDataset):
def __init__(self, root, split='train', mode='fine', target_type='instance', def __init__(self, root, split='train', mode='fine', target_type='instance',
transform=None, target_transform=None, transforms=None): transform=None, target_transform=None, transforms=None):
super(Cityscapes, self).__init__(root, transforms, transform, target_transform) 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.mode = 'gtFine' if mode == 'fine' else 'gtCoarse'
self.images_dir = os.path.join(self.root, 'leftImg8bit', split) self.images_dir = os.path.join(self.root, 'leftImg8bit', split)
self.targets_dir = os.path.join(self.root, self.mode, 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