Update: Replace the MEAN and STD params in dataset.

chairc · Nov 22, 2024 · c285830 · c285830
1 parent 12ed0df
commit c285830
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sr/dataset.py b/sr/dataset.py
@@ -12,7 +12,7 @@
 from PIL import Image
 from torch.utils.data import Dataset, DataLoader, DistributedSampler
 
-from config.setting import MEAN, STD
+from config.setting import SR_MEAN, SR_STD
 
 
 class SRDataset(Dataset):
@@ -28,7 +28,7 @@ def __init__(self, image_size=64, dataset_path="", scale=4):
             # To Tensor Format
             torchvision.transforms.ToTensor(),
             # For standardization, the mean and standard deviation
-            torchvision.transforms.Normalize(mean=MEAN, std=STD)
+            torchvision.transforms.Normalize(mean=SR_MEAN, std=SR_STD)
         ])
         self.hr_transforms = torchvision.transforms.Compose([
             # Resize input size
@@ -37,7 +37,7 @@ def __init__(self, image_size=64, dataset_path="", scale=4):
             # To Tensor Format
             torchvision.transforms.ToTensor(),
             # For standardization, the mean and standard deviation
-            torchvision.transforms.Normalize(mean=MEAN, std=STD)
+            torchvision.transforms.Normalize(mean=SR_MEAN, std=SR_STD)
         ])
 
     def __getitem__(self, index):