updated unpaired dataset

enhance_me/zero_dce/dataloader.py

@@ -1,6 +1,7 @@
 import tensorflow as tf
 from typing import List
 
+from ..commons import read_image
 from ..augmentation import UnpairedAugmentationFactory
 
 
@@ -17,15 +18,9 @@ class UnpairedLowLightDataset:
         self.apply_random_vertical_flip = apply_random_vertical_flip
         self.apply_random_rotation = apply_random_rotation
 
-    def _load_data(self, image_path):
-        image = tf.io.read_file(image_path)
-        image = tf.image.decode_png(image, channels=3)
-        image = tf.cast(image, dtype=tf.float32) / 255.0
-        return image
-
     def _get_dataset(self, images: List[str], batch_size: int, is_train: bool):
         dataset = tf.data.Dataset.from_tensor_slices((images))
-        dataset = dataset.map(self._load_data, num_parallel_calls=tf.data.AUTOTUNE)
+        dataset = dataset.map(read_image, num_parallel_calls=tf.data.AUTOTUNE)
         dataset = dataset.map(
             self.augmentation_factory.random_crop, num_parallel_calls=tf.data.AUTOTUNE
         )