Propagate value_range to RadomBrightness (#429)

keras-team · May 17, 2022 · 434fe01 · 434fe01
1 parent fd6f257
commit 434fe01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)