feature/load-dataset #2
@ -54,7 +54,7 @@ class Dataset:
|
||||
|
||||
def _get_image(self, image_path):
|
||||
img = tf.io.read_file(image_path)
|
||||
img = tf.io.decode_jpeg(img, channels=3) / 255
|
||||
img = tf.io.decode_jpeg(img, channels=3) / 255.
|
||||
return tf.cast(img, dtype=tf.float32, name=None)
|
||||
|
||||
def _preprocess(self, image_path):
|
||||
|
Loading…
Reference in New Issue
Block a user