NassimeBejaia commited on
Commit
5c0ed07
1 Parent(s): 73ac9a9

Update Utils.py

Browse files
Files changed (1) hide show
  1. Utils.py +1 -1
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