Update Utils.py

Utils.py

@@ -88,17 +88,25 @@ def decode_predictions(pred,greedy = True):
 #     return img
     
 
+# def encode_single_sample(path_dir):
+#     img = tf.io.read_file(path_dir)
+#     img = tf.io.decode_jpeg( img,  name=None)
+#     img.set_shape([img.shape[0], img.shape[1],img.shape[-1]])
+#     img = tf.image.rot90(img, k=1, name=None)
+#     rand = tf.random.uniform( (),   minval=0,maxval=1, dtype=tf.dtypes.float32,seed=None,name=None)
+#     img = tf.image.resize(img, [img_height, img_width])
+
+#     img=img/255.0
+
+#     return  img
+
 def encode_single_sample(path_dir):
     img = tf.io.read_file(path_dir)
-    img = tf.io.decode_jpeg( img,  name=None)
-    img.set_shape([img.shape[0], img.shape[1],img.shape[-1]])
+    img = tf.io.decode_jpeg(img, channels=3)  # Decode the image in RGB format
     img = tf.image.rot90(img, k=1, name=None)
-    rand = tf.random.uniform( (),   minval=0,maxval=1, dtype=tf.dtypes.float32,seed=None,name=None)
     img = tf.image.resize(img, [img_height, img_width])
-
-    img=img/255.0
-
-    return  img
+    img = img / 255.0  # Normalize the pixel values
+    return img