Пример #1
0
 def __init__(self, train=True,
              transform=Compose([ToFloat(), Normalize(mean=0.5, std=0.5)]),
              target_transform=None,
              label_type='fine'):
     assert label_type in ['fine', 'coarse']
     self.label_type = label_type
     super().__init__(train, transform, target_transform)
Пример #2
0
 def __init__(self,
              train=True,
              transform=Compose([Flatten(),
                                 ToFloat(),
                                 Normalize(0., 255.)]),
              target_transform=None):
     super().__init__(train, transform, target_transform)
Пример #3
0
 def __init__(
     self,
     train=True,
     transform=Compose([ToFloat(), Normalize(mean=0.5, std=0.5)]),
     target_transform=None,
 ):
     super().__init__(train, transform, target_transform)