From d7f27c9eb0ea5a2f5f2030da536d32389793dc7e Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 3 Jun 2024 08:47:20 +0200
Subject: [PATCH] [transforms] Move transforms to common lib
---
src/mednet/libs/classification/models/alexnet.py | 2 +-
src/mednet/libs/classification/models/densenet.py | 2 +-
src/mednet/libs/classification/models/pasa.py | 2 +-
src/mednet/libs/{classification => common}/models/transforms.py | 0
4 files changed, 3 insertions(+), 3 deletions(-)
rename src/mednet/libs/{classification => common}/models/transforms.py (100%)
diff --git a/src/mednet/libs/classification/models/alexnet.py b/src/mednet/libs/classification/models/alexnet.py
index 674046ca..5bc3f7da 100644
--- a/src/mednet/libs/classification/models/alexnet.py
+++ b/src/mednet/libs/classification/models/alexnet.py
@@ -13,9 +13,9 @@ import torchvision.models as models
import torchvision.transforms
from mednet.libs.common.data.typing import TransformSequence
from mednet.libs.common.models.model import Model
+from mednet.libs.common.models.transforms import RGB, SquareCenterPad
from .separate import separate
-from .transforms import RGB, SquareCenterPad
logger = logging.getLogger("mednet")
diff --git a/src/mednet/libs/classification/models/densenet.py b/src/mednet/libs/classification/models/densenet.py
index a4e94860..a2e9b61d 100644
--- a/src/mednet/libs/classification/models/densenet.py
+++ b/src/mednet/libs/classification/models/densenet.py
@@ -13,9 +13,9 @@ import torchvision.models as models
import torchvision.transforms
from mednet.libs.common.data.typing import TransformSequence
from mednet.libs.common.models.model import Model
+from mednet.libs.common.models.transforms import RGB, SquareCenterPad
from .separate import separate
-from .transforms import RGB, SquareCenterPad
logger = logging.getLogger("mednet")
diff --git a/src/mednet/libs/classification/models/pasa.py b/src/mednet/libs/classification/models/pasa.py
index e2e47010..555bd3fc 100644
--- a/src/mednet/libs/classification/models/pasa.py
+++ b/src/mednet/libs/classification/models/pasa.py
@@ -13,9 +13,9 @@ import torch.utils.data
import torchvision.transforms
from mednet.libs.common.data.typing import TransformSequence
from mednet.libs.common.models.model import Model
+from mednet.libs.common.models.transforms import Grayscale, SquareCenterPad
from .separate import separate
-from .transforms import Grayscale, SquareCenterPad
logger = logging.getLogger("mednet")
diff --git a/src/mednet/libs/classification/models/transforms.py b/src/mednet/libs/common/models/transforms.py
similarity index 100%
rename from src/mednet/libs/classification/models/transforms.py
rename to src/mednet/libs/common/models/transforms.py
--
GitLab