From a37f6bafa66f103397d6e1f2eb7739b9f66b6bf0 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Thu, 13 Jun 2024 17:15:24 +0200
Subject: [PATCH] [segmentation.models] Set default resize to 512 fo all models
---
src/mednet/libs/segmentation/config/models/driu.py | 2 +-
src/mednet/libs/segmentation/config/models/driu_bn.py | 2 +-
src/mednet/libs/segmentation/config/models/driu_od.py | 2 +-
src/mednet/libs/segmentation/config/models/driu_pix.py | 2 +-
src/mednet/libs/segmentation/config/models/hed.py | 3 +--
src/mednet/libs/segmentation/config/models/lwnet.py | 2 +-
src/mednet/libs/segmentation/config/models/m2unet.py | 3 +--
src/mednet/libs/segmentation/config/models/unet.py | 2 +-
src/mednet/libs/segmentation/models/driu.py | 2 +-
src/mednet/libs/segmentation/models/driu_bn.py | 2 +-
src/mednet/libs/segmentation/models/driu_od.py | 2 +-
src/mednet/libs/segmentation/models/driu_pix.py | 2 +-
src/mednet/libs/segmentation/models/hed.py | 2 +-
src/mednet/libs/segmentation/models/lwnet.py | 2 +-
src/mednet/libs/segmentation/models/m2unet.py | 2 +-
src/mednet/libs/segmentation/models/unet.py | 2 +-
16 files changed, 16 insertions(+), 18 deletions(-)
diff --git a/src/mednet/libs/segmentation/config/models/driu.py b/src/mednet/libs/segmentation/config/models/driu.py
index 847a7cd3..867619ea 100644
--- a/src/mednet/libs/segmentation/config/models/driu.py
+++ b/src/mednet/libs/segmentation/config/models/driu.py
@@ -39,5 +39,5 @@ model = DRIU(
amsbound=amsbound,
),
augmentation_transforms=[],
- crop_size=1024,
+ crop_size=512,
)
diff --git a/src/mednet/libs/segmentation/config/models/driu_bn.py b/src/mednet/libs/segmentation/config/models/driu_bn.py
index 8eddae7b..52fa78f4 100644
--- a/src/mednet/libs/segmentation/config/models/driu_bn.py
+++ b/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,
)
diff --git a/src/mednet/libs/segmentation/config/models/driu_od.py b/src/mednet/libs/segmentation/config/models/driu_od.py
index 0fd4b9f0..5994dff0 100644
--- a/src/mednet/libs/segmentation/config/models/driu_od.py
+++ b/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,
)
diff --git a/src/mednet/libs/segmentation/config/models/driu_pix.py b/src/mednet/libs/segmentation/config/models/driu_pix.py
index e0365c88..f9ddb848 100644
--- a/src/mednet/libs/segmentation/config/models/driu_pix.py
+++ b/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,
)
diff --git a/src/mednet/libs/segmentation/config/models/hed.py b/src/mednet/libs/segmentation/config/models/hed.py
index 0c88c9c2..d5d0f293 100644
--- a/src/mednet/libs/segmentation/config/models/hed.py
+++ b/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,
)
diff --git a/src/mednet/libs/segmentation/config/models/lwnet.py b/src/mednet/libs/segmentation/config/models/lwnet.py
index b0428f44..262636cc 100644
--- a/src/mednet/libs/segmentation/config/models/lwnet.py
+++ b/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,
)
diff --git a/src/mednet/libs/segmentation/config/models/m2unet.py b/src/mednet/libs/segmentation/config/models/m2unet.py
index bb360709..ec47f328 100644
--- a/src/mednet/libs/segmentation/config/models/m2unet.py
+++ b/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,
)
diff --git a/src/mednet/libs/segmentation/config/models/unet.py b/src/mednet/libs/segmentation/config/models/unet.py
index 10859a67..9919b12e 100644
--- a/src/mednet/libs/segmentation/config/models/unet.py
+++ b/src/mednet/libs/segmentation/config/models/unet.py
@@ -41,5 +41,5 @@ model = Unet(
amsbound=amsbound,
),
augmentation_transforms=[],
- crop_size=1024,
+ crop_size=512,
)
diff --git a/src/mednet/libs/segmentation/models/driu.py b/src/mednet/libs/segmentation/models/driu.py
index 6afab133..fc655cec 100644
--- a/src/mednet/libs/segmentation/models/driu.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/driu_bn.py b/src/mednet/libs/segmentation/models/driu_bn.py
index 3f8d8993..96b01990 100644
--- a/src/mednet/libs/segmentation/models/driu_bn.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/driu_od.py b/src/mednet/libs/segmentation/models/driu_od.py
index f2deb065..9272d391 100644
--- a/src/mednet/libs/segmentation/models/driu_od.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/driu_pix.py b/src/mednet/libs/segmentation/models/driu_pix.py
index 08f721d0..e84d45db 100644
--- a/src/mednet/libs/segmentation/models/driu_pix.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/hed.py b/src/mednet/libs/segmentation/models/hed.py
index 7ac25fb4..c20286fc 100644
--- a/src/mednet/libs/segmentation/models/hed.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/lwnet.py b/src/mednet/libs/segmentation/models/lwnet.py
index b9519e73..041b8ce3 100644
--- a/src/mednet/libs/segmentation/models/lwnet.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/m2unet.py b/src/mednet/libs/segmentation/models/m2unet.py
index 76042ba2..22ee9ee6 100644
--- a/src/mednet/libs/segmentation/models/m2unet.py
+++ b/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,
diff --git a/src/mednet/libs/segmentation/models/unet.py b/src/mednet/libs/segmentation/models/unet.py
index cc478a3c..f050e60c 100644
--- a/src/mednet/libs/segmentation/models/unet.py
+++ b/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,
--
GitLab