From 434fe0175b6d776f59bbff511820761fe550b7a2 Mon Sep 17 00:00:00 2001 From: bhack Date: Tue, 17 May 2022 07:17:10 +0200 Subject: [PATCH] Propagate value_range to RadomBrightness (#429) --- keras_cv/layers/preprocessing/rand_augment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_cv/layers/preprocessing/rand_augment.py b/keras_cv/layers/preprocessing/rand_augment.py index 75147b87c3..10f9ee8a06 100644 --- a/keras_cv/layers/preprocessing/rand_augment.py +++ b/keras_cv/layers/preprocessing/rand_augment.py @@ -139,7 +139,7 @@ def get_standard_policy(value_range, magnitude, magnitude_stddev, seed=None): color = cv_preprocessing.RandomColorDegeneration(**policy["color"], seed=seed) contrast = cv_preprocessing.RandomContrast(**policy["contrast"], seed=seed) brightness = cv_preprocessing.RandomBrightness( - **policy["brightness"], seed=seed + **policy["brightness"], value_range=value_range, seed=seed ) shear_x = cv_preprocessing.RandomShear(**policy["shear_x"], seed=seed) shear_y = cv_preprocessing.RandomShear(**policy["shear_y"], seed=seed)