[segmentation.models] Set default resize to 512 fo all models

src/mednet/libs/segmentation/config/models/driu.py

@@ -39,5 +39,5 @@ model = DRIU(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/driu_bn.py

@@ -39,5 +39,5 @@ model = DRIUBN(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/driu_od.py

@@ -39,5 +39,5 @@ model = DRIUOD(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/driu_pix.py

@@ -39,5 +39,5 @@ model = DRIUPix(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/hed.py

@@ -15,7 +15,6 @@ final_lr = 0.1
 gamma = 1e-3
 eps = 1e-8
 amsbound = False
-crop_size = 544
 
 model = HED(
     loss_type=MultiSoftJaccardBCELogitsLoss,
@@ -31,5 +30,5 @@ model = HED(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=crop_size,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/lwnet.py

@@ -22,5 +22,5 @@ model = LittleWNet(
     optimizer_type=Adam,
     optimizer_arguments=dict(lr=max_lr),
     augmentation_transforms=[],
-    crop_size=544,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/m2unet.py

@@ -28,7 +28,6 @@ final_lr = 0.1
 gamma = 1e-3
 eps = 1e-8
 amsbound = False
-crop_size = 544
 
 model = M2UNET(
     loss_type=SoftJaccardBCELogitsLoss,
@@ -44,5 +43,5 @@ model = M2UNET(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=crop_size,
+    crop_size=512,
 )

src/mednet/libs/segmentation/config/models/unet.py

@@ -41,5 +41,5 @@ model = Unet(
         amsbound=amsbound,
     ),
     augmentation_transforms=[],
-    crop_size=1024,
+    crop_size=512,
 )

src/mednet/libs/segmentation/models/driu.py

@@ -110,7 +110,7 @@ class DRIU(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 1024,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/driu_bn.py

@@ -113,7 +113,7 @@ class DRIUBN(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/driu_od.py

@@ -95,7 +95,7 @@ class DRIUOD(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/driu_pix.py

@@ -99,7 +99,7 @@ class DRIUPix(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/hed.py

@@ -113,7 +113,7 @@ class HED(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/lwnet.py

@@ -310,7 +310,7 @@ class LittleWNet(Model):
         optimizer_arguments: dict[str, typing.Any] = {},
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/m2unet.py

@@ -161,7 +161,7 @@ class M2UNET(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,

src/mednet/libs/segmentation/models/unet.py

@@ -102,7 +102,7 @@ class Unet(Model):
         augmentation_transforms: TransformSequence = [],
         num_classes: int = 1,
         pretrained: bool = False,
-        crop_size: int = 544,
+        crop_size: int = 512,
     ):
         super().__init__(
             loss_type,