diff --git a/keras/layers/preprocessing/image_preprocessing.py b/keras/layers/preprocessing/image_preprocessing.py index bcdd27924fd..eeed5bd77aa 100644 --- a/keras/layers/preprocessing/image_preprocessing.py +++ b/keras/layers/preprocessing/image_preprocessing.py @@ -1527,6 +1527,9 @@ def __init__(self, factor, value_range=(0, 255), seed=None, **kwargs): def augment_image(self, image, transformation=None): return self._brightness_adjust(image, transformation['rgb_delta']) + def augment_label(self, label, transformation=None): + return label + def get_random_transformation(self, image=None, label=None,