def __call__(self, image, target):
    if (random.random() < self.prob):
        image = F.hflip(image)
        target = target.transpose(0)
    return (image, target)
def __call__(self, image, target):
    if (random.random() < self.prob):
        image = F.hflip(image)
        target = target.transpose(0)
    return (image, target)