From 262b5a16e96448e08c422a5745f841888351b2eb Mon Sep 17 00:00:00 2001
From: Qihang Zhou <95111581+zqhang@users.noreply.github.com>
Date: Fri, 5 Jul 2024 20:20:20 +0800
Subject: [PATCH] Update mvtec.py

---
 generate_dataset_json/mvtec.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/generate_dataset_json/mvtec.py b/generate_dataset_json/mvtec.py
index 915c815..13eb913 100644
--- a/generate_dataset_json/mvtec.py
+++ b/generate_dataset_json/mvtec.py
@@ -25,13 +25,13 @@ def run(self):
                 for specie in species:
                     is_abnormal = True if specie not in ['good'] else False
                     img_names = os.listdir(f'{cls_dir}/{phase}/{specie}')
-                    mask_names = os.listdir(f'{cls_dir}/groundtruth/{specie}') if is_abnormal else None
+                    mask_names = os.listdir(f'{cls_dir}/ground_truth/{specie}') if is_abnormal else None
                     img_names.sort()
                     mask_names.sort() if mask_names is not None else None
                     for idx, img_name in enumerate(img_names):
                         info_img = dict(
                             img_path=f'{cls_name}/{phase}/{specie}/{img_name}',
-                            mask_path=f'{cls_name}/groundtruth/{specie}/{mask_names[idx]}' if is_abnormal else '',
+                            mask_path=f'{cls_name}/ground_truth/{specie}/{mask_names[idx]}' if is_abnormal else '',
                             cls_name=cls_name,
                             specie_name=specie,
                             anomaly=1 if is_abnormal else 0,