Method __call__

core/utils/flow_transforms.py:244–247

(self, inputs, target)

        self.augcolor = torchvision.transforms.ColorJitter(brightness=0.4, contrast=0.4, saturation=0.5, hue=0.5/3.14)

    def __call__(self, inputs, target):
        inputs[0] = np.asarray(self.augcolor(Image.fromarray(np.uint8(inputs[0]*255))))/255.
        inputs[1] = np.asarray(self.augcolor(Image.fromarray(np.uint8(inputs[1]*255))))/255.
        return inputs,target


class PCAAug(object):