Skip to content
Snippets Groups Projects
Commit d7f27c9e authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

[transforms] Move transforms to common lib

parent 026f3214
No related branches found
No related tags found
1 merge request!46Create common library
......@@ -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")
......
......@@ -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")
......
......@@ -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")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment