Spaces:
Runtime error
Runtime error
File size: 420 Bytes
fc8c192 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from paddle.vision.transforms import ColorJitter as pp_ColorJitter
__all__ = ["ColorJitter"]
class ColorJitter(object):
def __init__(self, brightness=0, contrast=0, saturation=0, hue=0, **kwargs):
self.aug = pp_ColorJitter(brightness, contrast, saturation, hue)
def __call__(self, data):
image = data["image"]
image = self.aug(image)
data["image"] = image
return data
|