From 09484ff5a48c2b68c2eaa7df451f93ab982a6592 Mon Sep 17 00:00:00 2001
From: dummyindex <dummtindex@gmail.com>
Date: Thu, 28 Mar 2024 03:28:45 -0400
Subject: [PATCH] update csn_configs.py: modify augV4

---
 livecellx/model_zoo/segmentation/csn_configs.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/livecellx/model_zoo/segmentation/csn_configs.py b/livecellx/model_zoo/segmentation/csn_configs.py
index 6362546..3a793b3 100644
--- 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