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
from ..data.typing import TransformSequence
from .separate import separate
from .transforms import RGB
from .transforms import RGB, SquareCenterPad
from .typing import Checkpoint
logger = logging.getLogger(__name__)
@@ -74,6 +74,7 @@ class Alexnet(pl.LightningModule):
self.num_classes = num_classes
self.model_transforms = [
SquareCenterPad(),
torchvision.transforms.Resize(512, antialias=True),
RGB(),
]
Loading