Spaces:
Runtime error
Runtime error
Commit
•
5c0ed07
1
Parent(s):
73ac9a9
Update Utils.py
Browse files
Utils.py
CHANGED
@@ -81,7 +81,7 @@ def encode_single_sample(path_dir, label=None):
|
|
81 |
img.set_shape([img.shape[0], img.shape[1], img.shape[-1]])
|
82 |
img = tf.image.rot90(img, k=1, name=None)
|
83 |
img = tf.image.resize(img, [img_height, img_width])
|
84 |
-
img = tf.image.rgb_to_grayscale(img) # Convert image to grayscale
|
85 |
img = img/255.0
|
86 |
return img
|
87 |
|
|
|
81 |
img.set_shape([img.shape[0], img.shape[1], img.shape[-1]])
|
82 |
img = tf.image.rot90(img, k=1, name=None)
|
83 |
img = tf.image.resize(img, [img_height, img_width])
|
84 |
+
# img = tf.image.rgb_to_grayscale(img) # Convert image to grayscale
|
85 |
img = img/255.0
|
86 |
return img
|
87 |
|