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.
34 lines
1.2 KiB
Plaintext
34 lines
1.2 KiB
Plaintext
from torchvision.transforms.v2 import Compose, RandomAdjustSharpness, RandomHorizontalFlip, Resize
|
|
|
|
train_transform = Compose(
|
|
[
|
|
RandomAdjustSharpness(sharpness_factor=0.7, p=0.5),
|
|
RandomHorizontalFlip(p=0.5),
|
|
Resize((256, 256), antialias=True),
|
|
Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|
|
],
|
|
)
|
|
eval_transform = Compose(
|
|
[
|
|
Resize((256, 256), antialias=True),
|
|
Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|
|
],
|
|
)
|
|
|
|
datamodule = MVTecAD(train_transform=train_transform, eval_transform=eval_transform)
|
|
datamodule.prepare_data()
|
|
datamodule.setup()
|
|
|
|
datamodule.train_transform
|
|
# Compose(
|
|
# RandomAdjustSharpness(p=0.5, sharpness_factor=0.7)
|
|
# RandomHorizontalFlip(p=0.5)
|
|
# Resize(size=[256, 256], interpolation=InterpolationMode.BILINEAR, antialias=True)
|
|
# Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225], inplace=False)
|
|
# )
|
|
datamodule.eval_transform
|
|
# Compose(
|
|
# Resize(size=[256, 256], interpolation=InterpolationMode.BILINEAR, antialias=True)
|
|
# Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225], inplace=False)
|
|
# )
|