diff --git a/src/mednet/libs/classification/models/alexnet.py b/src/mednet/libs/classification/models/alexnet.py index 674046ca050d695ff3c81916e1e8fc2b31fe0367..5bc3f7dae06e6fa147bc6e6a07b051204f2d4223 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 a4e948609cb5f4273fab491985d2d85c7339000c..a2e9b61d585ad4a0a1a31e6ec86e45dc54e989f3 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 e2e47010ed431fc8436591c264bb7911531d5f52..555bd3fc188b778f2e6d43bab76358ef1b9e8255 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