Merge pull request #89 from KarhouTam/dev

fix(data): Make normalization works correctly (#88)
KarhouTam · Sep 15, 2024 · bff79ed · bff79ed
2 parents 21e6a33 + 7d24cb0
commit bff79ed
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/data/utils/datasets.py b/data/utils/datasets.py
@@ -42,9 +42,8 @@ def __init__(
 
     def _rescale_data(self):
         max_val = self.data.max()
-        min_val = self.data.min()
-        if max_val > 1.0 or min_val < 0.0:
-            self.data = (self.data - min_val) / (max_val - min_val)
+        if max_val > 1.0:
+            self.data /= 255.0
 
     def __getitem__(self, index):
         data, targets = self.data[index], self.targets[index]