diff --git a/src/patchcore/datasets/mvtec.py b/src/patchcore/datasets/mvtec.py index 54c88b5..fb67205 100644 --- a/src/patchcore/datasets/mvtec.py +++ b/src/patchcore/datasets/mvtec.py @@ -88,6 +88,9 @@ def __init__( self.imagesize = (3, imagesize, imagesize) + self.transform_std = IMAGENET_STD + self.transform_mean = IMAGENET_MEAN + def __getitem__(self, idx): classname, anomaly, image_path, mask_path = self.data_to_iterate[idx] image = PIL.Image.open(image_path).convert("RGB")