feat: closes #24

2023-10-03 19:02:02 +01:00
parent f68f0e2444
commit 84b9c40a72
4 changed files with 26 additions and 38 deletions
--- a/views/py/python_model_template.py
+++ b/views/py/python_model_template.py
@@ -32,10 +32,7 @@ def pathToLabel(path):
  #return tf.strings.as_string([path])

 def decode_image(img):
-  # channels were reduced to 1 since image is grayscale
-  # TODO chnage channel number based if grayscale
-  img = tf.io.decode_png(img, channels=3)
-
+  img = tf.io.decode_png(img, channels={{.ColorMode}})
  return tf.image.resize(img, image_size)

 def process_path(path):