geekyrakshit commited on
Commit
d7c18de
·
1 Parent(s): 0668e89

bux fix in unpaired dataloader

Browse files
Files changed (1) hide show
  1. enhance_me/zero_dce/dataloader.py +1 -1
enhance_me/zero_dce/dataloader.py CHANGED
@@ -20,7 +20,7 @@ class UnpairedLowLightDataset:
20
  def _load_data(self, image_path):
21
  image = tf.io.read_file(image_path)
22
  image = tf.image.decode_png(image, channels=3)
23
- image = image / 255.0
24
  return image
25
 
26
  def _get_dataset(self, images: List[str], batch_size: int, is_train: bool):
 
20
  def _load_data(self, image_path):
21
  image = tf.io.read_file(image_path)
22
  image = tf.image.decode_png(image, channels=3)
23
+ image = tf.cast(image, dtype=tf.float32) / 255.0
24
  return image
25
 
26
  def _get_dataset(self, images: List[str], batch_size: int, is_train: bool):