In mvtec.py, assigning transform_std and transform_mean to solve Issue …

…amazon-science#80 in patchcore-inspection.
rovodanica · Jan 24, 2024 · 1264a1f · 1264a1f
1 parent ec1960c
commit 1264a1f
Show file tree

Hide file tree

Showing 16 changed files with 3 additions and 0 deletions.
diff --git a/.tmp.driveupload/17471 b/.tmp.driveupload/17471
diff --git a/.tmp.driveupload/17475 b/.tmp.driveupload/17475
diff --git a/.tmp.driveupload/17479 b/.tmp.driveupload/17479
diff --git a/.tmp.driveupload/17483 b/.tmp.driveupload/17483
diff --git a/.tmp.driveupload/17487 b/.tmp.driveupload/17487
diff --git a/.tmp.driveupload/17491 b/.tmp.driveupload/17491
diff --git a/.tmp.driveupload/17495 b/.tmp.driveupload/17495
diff --git a/.tmp.driveupload/17499 b/.tmp.driveupload/17499
diff --git a/.tmp.driveupload/17503 b/.tmp.driveupload/17503
diff --git a/.tmp.driveupload/17507 b/.tmp.driveupload/17507
diff --git a/.tmp.driveupload/17511 b/.tmp.driveupload/17511
diff --git a/.tmp.driveupload/17515 b/.tmp.driveupload/17515
diff --git a/.tmp.driveupload/17536 b/.tmp.driveupload/17536
diff --git a/.tmp.driveupload/17538 b/.tmp.driveupload/17538
diff --git a/.tmp.driveupload/17542 b/.tmp.driveupload/17542
diff --git 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")