From dbaf637de75d3a5758c63a0b546bede40194c845 Mon Sep 17 00:00:00 2001 From: Sarvagya Malaviya Date: Thu, 24 Mar 2022 21:07:16 +0530 Subject: [PATCH] Vectorizing Equalization --- keras_cv/layers/preprocessing/equalization.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/keras_cv/layers/preprocessing/equalization.py b/keras_cv/layers/preprocessing/equalization.py index 3c57789c6e..5151a4b859 100644 --- a/keras_cv/layers/preprocessing/equalization.py +++ b/keras_cv/layers/preprocessing/equalization.py @@ -45,7 +45,10 @@ class Equalization(tf.keras.__internal__.layers.BaseImageAugmentationLayer): def __init__(self, bins=256, value_range=(0, 255), **kwargs): super().__init__(**kwargs) self.bins = bins +<<<<<<< HEAD self.value_range = value_range +======= +>>>>>>> Vectorizing Equalization @tf.function def equalize_channel(self, image, channel_index):