mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
18 lines
541 B
Plaintext
18 lines
541 B
Plaintext
from torchvision.transforms.v2 import Resize
|
|
|
|
transform = Resize((256, 256))
|
|
datamodule = MVTecAD(transform=transform)
|
|
|
|
datamodule.prepare_data()
|
|
datamodule.setup()
|
|
|
|
datamodule.train_transform
|
|
# Resize(size=[256, 256], interpolation=InterpolationMode.BILINEAR, antialias=warn)
|
|
datamodule.eval_transform
|
|
# Resize(size=[256, 256], interpolation=InterpolationMode.BILINEAR, antialias=warn)
|
|
|
|
next(iter(datamodule.train_data))["image"].shape
|
|
# torch.Size([3, 256, 256])
|
|
next(iter(datamodule.test_data))["image"].shape
|
|
# torch.Size([3, 256, 256])
|