Mikalai Rahachou mickolka
jimfleming
/ colorize.py
Last active
November 25, 2022 19:36
A utility function for TensorFlow that maps a grayscale image to a matplotlib colormap for use with TensorBoard image summaries.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import matplotlib | |
| import matplotlib.cm | |
| import tensorflow as tf | |
| def colorize(value, vmin=None, vmax=None, cmap=None): | |
| """ | |
| A utility function for TensorFlow that maps a grayscale image to a matplotlib | |
| colormap for use with TensorBoard image summaries. |