data norm.

Author: likyoo

change_detection_pytorch/datasets/LEVIR_CD.py

@@ -19,7 +19,7 @@ def get_default_transform(self):
         default_transform = A.Compose([
             A.RandomCrop(self.size, self.size),
             # A.ShiftScaleRotate(),
-            A.Normalize(mean=(0, 0, 0), std=(1, 1, 1)),  # div(255)
+            A.Normalize(),
             ToTensorV2()
         ], additional_targets={'image_2': 'image'})
         return default_transform
@@ -30,7 +30,7 @@ def get_test_transform(self):
         from change_detection_pytorch.datasets.transforms.albu import (
             ChunkImage, ToTensorTest)
         test_transform = A.Compose([
-            A.Normalize(mean=(0, 0, 0), std=(1, 1, 1)),  # div(255)
+            A.Normalize(),
             ChunkImage(self.size),
             ToTensorTest(),
         ], additional_targets={'image_2': 'image'})

change_detection_pytorch/datasets/SVCD.py

@@ -20,7 +20,7 @@ def get_default_transform(self):
             A.Resize(self.size, self.size),
             # A.HorizontalFlip(p=0.5),
             # A.RandomRotate90(p=0.5),
-            A.Normalize(mean=(0, 0, 0), std=(1, 1, 1)),     # div(255)
+            A.Normalize(),
             ToTensorV2()
         ], additional_targets={'image_2': 'image'})
         return default_transform
@@ -30,7 +30,7 @@ def get_test_transform(self):
 
         test_transform = A.Compose([
             A.Resize(self.size, self.size),
-            A.Normalize(mean=(0, 0, 0), std=(1, 1, 1)),     # div(255)
+            A.Normalize(),
             ToTensorV2()
         ], additional_targets={'image_2': 'image'})
         return test_transform