Spaces:
Runtime error
Runtime error
Commit
·
d7c18de
1
Parent(s):
0668e89
bux fix in unpaired dataloader
Browse files
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):
|