Update jnd_dataset.py

richzhang · Dec 19, 2023 · ab167cb · ab167cb
1 parent 1fd3552
commit ab167cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/dataset/jnd_dataset.py b/data/dataset/jnd_dataset.py
@@ -21,7 +21,7 @@ def initialize(self, dataroot, load_size=64):
  self.p1_paths = sorted(self.p1_paths)
 
  transform_list = []
- transform_list.append(transforms.Scale(load_size))
+ transform_list.append(transforms.Resize(load_size))
  transform_list += [transforms.ToTensor(),
  transforms.Normalize((0.5, 0.5, 0.5),(0.5, 0.5, 0.5))]