feat: closes #23

2023-10-03 18:49:36 +01:00
parent 884a978e3b
commit f68f0e2444
4 changed files with 24 additions and 10 deletions
--- a/views/py/python_model_template.py
+++ b/views/py/python_model_template.py
@@ -34,7 +34,7 @@ def pathToLabel(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=1)
+  img = tf.io.decode_png(img, channels=3)

  return tf.image.resize(img, image_size)