Merge pull request #218 from isl-org/dvdhfnr/load-image

simplify load image
isl-org · Sep 10, 2021 · ac40bc3 · ac40bc3
2 parents be16fc9 + 8e04fef
commit ac40bc3
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/policy/openbot/utils.py b/policy/openbot/utils.py
@@ -163,12 +163,9 @@ def list_dirs(path):
 
 
 def load_img(file_path):
-    # load the raw data from the file as a string
     img = tf.io.read_file(file_path)
-    # convert the compressed string to a 3D uint8 tensor
-    img = tf.image.decode_jpeg(img, channels=3)
-    # Use `convert_image_dtype` to convert to floats in the [0,1] range.
-    img = tf.image.convert_image_dtype(img, tf.float32)
+    img = tf.image.decode_image(img, channels=3, dtype=tf.float32)
+
     return img