Added normalize_patches()

dataset.py

@@ -11,6 +11,15 @@ from time import time
 import gc
 #from torch.utils.data import TensorDataset,Dataloader 
 
+def normalize_patches(patches):
+    normalized_patches = []
+    for patch in patches:
+        # normalize the patch by subtracting the mean and dividing by the standard deviation
+        normalized_patch = (patch - patch.mean()) / patch.std()
+        normalized_patches.append(normalized_patch)
+return normalized_patches
+
+
 def connected_components_torch(images,crop_size=15,NrPixels = 2048):
     window=int(crop_size/2)
     start = torch.cuda.Event(enable_timing=True)