search

MiniBullLab / easy_ai

3 stars 1 forks source link

one class test报错 #133

Closed foww-0001 closed 3 years ago

foww-0001 commented 3 years ago

运行命令：

python3 easyai/test_task.py -t one_class -i /home/wfw/data/VOCdevkit/MVTec/bottle/ImageSets/val.txt -m PatchCoreNet -w  /home/wfw/workspace/HASCO/all_wights/wide_resnet50_2_patchcore.pt

报错如下：

2021-08-12 17:20:53,554 ERROR   [test_task.py, 37] Traceback (most recent call last):
  File "easyai/test_task.py", line 35, in test
    task.process_test(self.val_path)
  File "/home/wfw/workspace/Test/easy_ai/easyai/tasks/one_class/one_class_test.py", line 30, in process_test
    self.test(epoch)
  File "/home/wfw/workspace/Test/easy_ai/easyai/tasks/one_class/one_class_test.py", line 35, in test
    result, _ = self.inference.result_process.post_process(prediction)
  File "/home/wfw/workspace/Test/easy_ai/easyai/tasks/one_class/one_class_result_process.py", line 18, in post_process
    class_indices, class_confidence = self.process_func(prediction)
  File "/home/wfw/workspace/Test/easy_ai/easyai/tasks/one_class/post_process/patch_core_post_process.py", line 66, in __call__
    embedding_test = np.array(self.reshape_embedding(np.array(prediction)))
  File "/home/wfw/workspace/Test/easy_ai/easyai/tasks/one_class/post_process/patch_core_post_process.py", line 43, in reshape_embedding
    for j in range(embedding.shape[3]):
IndexError: tuple index out of range

2021-08-12 17:20:53,554 ERROR   [test_task.py, 38] tuple index out of range