modify

fu123456 · Sep 19, 2023 · 5f46c99 · 5f46c99
1 parent 37effb6
commit 5f46c99
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/utils/data_loader_four_tuples_mix.py b/utils/data_loader_four_tuples_mix.py
@@ -77,14 +77,10 @@ def __call__(self, img):
 class ImageTransform():
  def __init__(self, size=256, crop_size=256, mean=(0.5, ), std=(0.5, )):
  self.data_transform = {'train': four_tuple_data_processing.Compose([four_tuple_data_processing.Scale(size=size),
- four_tuple_data_processing.RandomCrop(size=crop_size),
- four_tuple_data_processing.RandomHorizontalFlip(p=0.5),
  four_tuple_data_processing.ToTensor(),
  four_tuple_data_processing.Normalize(mean, std)]),
 
  'test': four_tuple_data_processing.Compose([four_tuple_data_processing.Scale(size=size),
- four_tuple_data_processing.RandomCrop(size=crop_size),
- four_tuple_data_processing.RandomHorizontalFlip(p=0.5),
  four_tuple_data_processing.ToTensor(),
  four_tuple_data_processing.Normalize(mean, std)])}
 

diff --git a/utils/data_loader_seven_tuples.py b/utils/data_loader_seven_tuples.py
@@ -95,14 +95,10 @@ def __call__(self, img):
 class ImageTransform():
  def __init__(self, size=256, crop_size=256, mean=(0.5, ), std=(0.5, )):
  self.data_transform = {'train': seven_tuple_data_processing.Compose([seven_tuple_data_processing.Scale(size=size),
- seven_tuple_data_processing.RandomCrop(size=crop_size),
- seven_tuple_data_processing.RandomHorizontalFlip(p=0.5),
  seven_tuple_data_processing.ToTensor(),
  seven_tuple_data_processing.Normalize(mean, std)]),
 
  'test': seven_tuple_data_processing.Compose([seven_tuple_data_processing.Scale(size=size),
- seven_tuple_data_processing.RandomCrop(size=crop_size),
- seven_tuple_data_processing.RandomHorizontalFlip(p=0.5),
  seven_tuple_data_processing.ToTensor(),
  seven_tuple_data_processing.Normalize(mean, std)])}