From b33622885e09bd861418c8cd574494e726ddc773 Mon Sep 17 00:00:00 2001 From: Yunchu Lee Date: Fri, 30 Aug 2024 14:58:42 +0900 Subject: [PATCH] fix wrong template id mapping for anomaly --- src/otx/tools/converter.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/otx/tools/converter.py b/src/otx/tools/converter.py index 5b19febd73d..e7f8e980855 100644 --- a/src/otx/tools/converter.py +++ b/src/otx/tools/converter.py @@ -122,29 +122,29 @@ }, # ANOMALY_CLASSIFICATION "ote_anomaly_classification_padim": { - "task": OTXTaskType.SEMANTIC_SEGMENTATION, + "task": OTXTaskType.ANOMALY_CLASSIFICATION, "model_name": "padim", }, "ote_anomaly_classification_stfpm": { - "task": OTXTaskType.SEMANTIC_SEGMENTATION, + "task": OTXTaskType.ANOMALY_CLASSIFICATION, "model_name": "stfpm", }, # ANOMALY_DETECTION "ote_anomaly_detection_padim": { - "task": OTXTaskType.SEMANTIC_SEGMENTATION, + "task": OTXTaskType.ANOMALY_DETECTION, "model_name": "padim", }, "ote_anomaly_detection_stfpm": { - "task": OTXTaskType.SEMANTIC_SEGMENTATION, + "task": OTXTaskType.ANOMALY_DETECTION, "model_name": "stfpm", }, # ANOMALY_SEGMENTATION "ote_anomaly_segmentation_padim": { - "task": OTXTaskType.SEMANTIC_SEGMENTATION, + "task": OTXTaskType.ANOMALY_SEGMENTATION, "model_name": "padim", }, "ote_anomaly_segmentation_stfpm": { - "task": OTXTaskType.SEMANTIC_SEGMENTATION, + "task": OTXTaskType.ANOMALY_SEGMENTATION, "model_name": "stfpm", }, }