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 ), ])
def __init__(self, cropsize): self.trans_func = T.Compose([ T.RandomResizedCrop((1, 1), cropsize), ])