def __init__(self, mean, std): self.transform = Normalize_th(mean, std)
def __init__(self, mean, std): self.transform = Normalize_th([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])