From 3c4ae87d9a8a39d58098d86d71c303160e621935 Mon Sep 17 00:00:00 2001 From: Naomi-Nour Date: Tue, 16 Sep 2025 15:53:54 +0100 Subject: [PATCH] Fix missing transform_std and transform_mean attributes in MVTecDataset - Adds missing transform_std and transform_mean attributes to MVTecDataset class - Fixes AttributeError when using --save_segmentation_images flag - Resolves issues #15, #35, and #80 - Uses existing ImageNet normalization constants - Maintains backward compatibility --- src/patchcore/datasets/mvtec.py | 3 +++ 1 file changed, 3 insertions(+) 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")