update csn_configs.py: modify augV4

xing-lab-pitt · Mar 28, 2024 · 09484ff · 09484ff
1 parent 14c94f5
commit 09484ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/livecellx/model_zoo/segmentation/csn_configs.py b/livecellx/model_zoo/segmentation/csn_configs.py
@@ -170,7 +170,8 @@ def gen_train_transform_v4(
             transforms.RandomVerticalFlip(),
             transforms.RandomAffine(degrees=degrees, translate=translation_range, scale=scale, shear=10),
             gauss_noise_tensor,
-            transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
+            transforms.Resize((412, 412)),
+            transforms.Normalize([0.485], [0.229]),
         ]
     )
     return train_transforms