Add tensorflow function to read images

sawubona-repo · Sep 14, 2019 · 837b164 · 837b164
1 parent 15c2dab
commit 837b164
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/notebooks/utils.py b/notebooks/utils.py
@@ -60,6 +60,12 @@ def get_scaled_size() -> tf.Tensor:
     return tf.cast(rescaled_image, dtype)
 
 
+@tf.function
+def read_image(filename: str, **kwargs) -> tf.Tensor:
+    stream = tf.io.read_file(filename)
+    return tf.image.decode_image(stream, **kwargs)
+
+
 def show_images(images: list, **kwargs):
     fig, axs = plt.subplots(1, len(images), figsize=(19, 10))
     for image, ax in zip(images, axs):