Esempio n. 1
0
 def __init__(self, scales, cropsize):
     self.trans_func = T.Compose([
         T.RandomResizedCrop(scales, cropsize),
         T.RandomHorizontalFlip(),
         T.ColorJitter(
             brightness=0.4,
             contrast=0.4,
             saturation=0.4
         ),
     ])
Esempio n. 2
0
 def __init__(self, cropsize):
     self.trans_func = T.Compose([
         T.RandomResizedCrop((1, 1), cropsize),
     ])