Test

inaturalist · Feb 15, 2024 · 6ade4fb · 6ade4fb
1 parent 72922bd
commit 6ade4fb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/inat_inferrer.py b/lib/inat_inferrer.py
@@ -91,7 +91,9 @@ def prepare_image_for_inference(self, file_path):
 
         filename = os.path.basename(file_path)
         output_file_path = '/home/inaturalist/vision/save/' + filename
-        image_bytes = tf.io.encode_jpeg(image)
+
+        resized_image_uint8 = tf.cast(image, tf.uint8)
+        image_bytes = tf.io.encode_jpeg(resized_image_uint8)
         with tf.io.gfile.GFile(output_file_path, 'wb') as f:
             f.write(image_bytes.numpy())