Skip to content

Instantly share code, notes, and snippets.

@alegoritma

alegoritma/tf_stretch_histogram.py

Created March 12, 2021 12:41
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save alegoritma/4227282cbda3a00d6ba07a105bea78f4 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save alegoritma/4227282cbda3a00d6ba07a105bea78f4 to your computer and use it in GitHub Desktop.
Download ZIP
tf.stretch_histogram
Raw
tf_stretch_histogram.py
@tf.function
def tf_strech_histogram(image):
rmax = tf.reduce_max(image, axis=[-3,-2])
rmin = tf.reduce_min(image, axis=[-3,-2])
return (image-rmin)/(rmax-rmin)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment