From a2374373fee646a9341946ccab53a9b078f57440 Mon Sep 17 00:00:00 2001 From: Ugeun Park <37043543+shashaka@users.noreply.github.com> Date: Sat, 14 Dec 2024 03:34:19 +0900 Subject: [PATCH] Update example and logic for mix_up (#20643) --- .../layers/preprocessing/image_preprocessing/random_hue.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_hue.py b/keras/src/layers/preprocessing/image_preprocessing/random_hue.py index 666fd3d2efd..29ec528e41e 100644 --- a/keras/src/layers/preprocessing/image_preprocessing/random_hue.py +++ b/keras/src/layers/preprocessing/image_preprocessing/random_hue.py @@ -55,7 +55,8 @@ class RandomHue(BaseImagePreprocessingLayer): ```python (images, labels), _ = keras.datasets.cifar10.load_data() random_hue = keras.layers.RandomHue(factor=0.5, value_range=[0, 1]) - augmented_images_batch = random_hue(images[:32]) + images = keras.ops.cast(images, "float32") + augmented_images_batch = random_hue(images[:8]) ``` """ @@ -90,7 +91,7 @@ def get_random_transformation(self, data, training=True, seed=None): if seed is None: seed = self._get_seed_generator(self.backend._backend) - invert = self.backend.random.uniform((1,), seed=seed) + invert = self.backend.random.uniform((batch_size,), seed=seed) invert = self.backend.numpy.where( invert > 0.5,