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