div by float

2024-05-05 12:44:54 +02:00 · 2024-05-05 12:44:54 +02:00 · b7ca0fae45
parent de27695d53
commit b7ca0fae45
1 changed files with 1 additions and 1 deletions
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@ -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):