Vectorize Equalization

keras-team · Mar 29, 2022 · 7e52368 · 7e52368
1 parent a7105fb
commit 7e52368
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, bins=256, value_range=(0, 255), **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.