Skip to content
Snippets Groups Projects

Make square centre-padding a model transform

Merged André Anjos requested to merge issue-23-and-39 into main
4 files
+ 52
6
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -15,7 +15,7 @@ import torchvision.transforms
@@ -15,7 +15,7 @@ import torchvision.transforms
from ..data.typing import TransformSequence
from ..data.typing import TransformSequence
from .separate import separate
from .separate import separate
from .transforms import RGB
from .transforms import RGB, SquareCenterPad
from .typing import Checkpoint
from .typing import Checkpoint
logger = logging.getLogger(__name__)
logger = logging.getLogger(__name__)
@@ -74,6 +74,7 @@ class Alexnet(pl.LightningModule):
@@ -74,6 +74,7 @@ class Alexnet(pl.LightningModule):
self.num_classes = num_classes
self.num_classes = num_classes
self.model_transforms = [
self.model_transforms = [
 
SquareCenterPad(),
torchvision.transforms.Resize(512, antialias=True),
torchvision.transforms.Resize(512, antialias=True),
RGB(),
RGB(),
]
]
Loading