Skip to content

Commit 4c05e0c

Browse files
committed
Fix issues with randomgrayscale layer
1 parent d96f910 commit 4c05e0c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

keras/src/layers/preprocessing/image_preprocessing/random_grayscale.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ class RandomGrayscale(BaseImagePreprocessingLayer):
4545
will have the same value.
4646
"""
4747

48-
def __init__(self, factor=0.5, data_format=None, **kwargs):
48+
def __init__(self, factor=0.5, data_format=None, seed=None, **kwargs):
4949
super().__init__(**kwargs)
5050
if factor < 0 or factor > 1:
5151
raise ValueError(
@@ -54,7 +54,8 @@ def __init__(self, factor=0.5, data_format=None, **kwargs):
5454
)
5555
self.factor = factor
5656
self.data_format = backend.standardize_data_format(data_format)
57-
self.generator = self.backend.random.SeedGenerator()
57+
self.seed = seed
58+
self.generator = self.backend.random.SeedGenerator(seed)
5859

5960
def get_random_transformation(self, images, training=True, seed=None):
6061
if seed is None:

0 commit comments

Comments
 (0)