From b7ca0fae45b2056ae3fb54b663755575f2142a64 Mon Sep 17 00:00:00 2001
From: mszmyd <erilf9@gmail.com>
Date: Sun, 5 May 2024 12:44:54 +0200
Subject: [PATCH] div by float

---
 dataset/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dataset/dataset.py b/dataset/dataset.py
index d098ead..6b4489b 100644
--- 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):