fffa476e1e50ff0a59700a6fbfd3531d62fe692a,kornia/augmentation/augmentations.py,RandomVerticalFlip,forward,#RandomVerticalFlip#Any#Any#,154

Before Change

        super(RandomVerticalFlip, self).__init__(p=p, return_transform=return_transform)

    def forward(self, input: UnionType, params: Optional[Dict[str, torch.Tensor]] = None) -> UnionType:  // type: ignore
        return self.forward_flip(input, F.apply_vflip, params)


class ColorJitter(AugmentationBase):

After Change

        return pg._random_prob_gen(batch_size, p)

    def forward(self, input: UnionType, params: Optional[Dict[str, torch.Tensor]] = None) -> UnionType:  // type: ignore
        if params is None:
            batch_size = self.infer_batch_size(input)
            params: Dict[str, torch.Tensor] = self.get_params(batch_size, self.p)
        return super().forward(input, params)


class ColorJitter(AugmentationBase):

Instances