Skip to content

Commit

Permalink
Vectorizing Equalization
Browse files Browse the repository at this point in the history
  • Loading branch information
quantumalaviya committed Mar 29, 2022
1 parent a1f2fa6 commit dbaf637
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions keras_cv/layers/preprocessing/equalization.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit dbaf637

Please sign in to comment.