From 020f3395244d48233b806c78e72bbb1104526a01 Mon Sep 17 00:00:00 2001 From: Sarvagya Malaviya Date: Tue, 29 Mar 2022 21:03:59 +0530 Subject: [PATCH] Vectorize Equalization --- keras_cv/layers/preprocessing/equalization.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/keras_cv/layers/preprocessing/equalization.py b/keras_cv/layers/preprocessing/equalization.py index c7299a518b..fcc7d2b314 100644 --- a/keras_cv/layers/preprocessing/equalization.py +++ b/keras_cv/layers/preprocessing/equalization.py @@ -45,10 +45,7 @@ class Equalization(tf.keras.__internal__.layers.BaseImageAugmentationLayer): def __init__(self, value_range, bins=256, **kwargs): super().__init__(**kwargs) self.bins = bins -<<<<<<< HEAD self.value_range = value_range -======= ->>>>>>> Vectorizing Equalization @tf.function def equalize_channel(self, image, channel_index):