Bugfix: Data Loader

animikhaich · animikhaich · commit 75b1c5834514 · 2021-04-03T18:25:02.000+05:30
diff --git a/utils/data_loader.py b/utils/data_loader.py
@@ -106,8 +106,8 @@ def load_image(self, file_path) -> tuple:
         img = tf.io.decode_image(
             img, channels=self.NUM_CHANNELS, expand_animations=False
         )
-        img = tf.image.resize(img, [self.HEIGHT, self.WIDTH])
         img = tf.image.convert_image_dtype(img, tf.float32)
+        img = tf.image.resize(img, [self.HEIGHT, self.WIDTH])
 
         return img, tf.cast(label, tf.float32)
 

Original file line number	Diff line number	Diff line change
`@@ -106,8 +106,8 @@ def load_image(self, file_path) -> tuple:`
`106`	`106`	`img = tf.io.decode_image(`
`107`	`107`	`img, channels=self.NUM_CHANNELS, expand_animations=False`
`108`	`108`	`)`
`109`		`- img = tf.image.resize(img, [self.HEIGHT, self.WIDTH])`
`110`	`109`	`img = tf.image.convert_image_dtype(img, tf.float32)`
	`110`	`+ img = tf.image.resize(img, [self.HEIGHT, self.WIDTH])`
`111`	`111`
`112`	`112`	`return img, tf.cast(label, tf.float32)`
`113`	`113`