Vectorize Equalization

keras-team · Apr 2, 2022 · e5fab4b · e5fab4b
1 parent 020f339
commit e5fab4b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/keras_cv/layers/preprocessing/equalization.py b/keras_cv/layers/preprocessing/equalization.py
@@ -47,7 +47,6 @@ def __init__(self, value_range, bins=256, **kwargs):
         self.bins = bins
         self.value_range = value_range
 
-    @tf.function
     def equalize_channel(self, image, channel_index):
         """equalize_channel performs histogram equalization on a single channel.