Fix wrong template id mapping for anomaly (#3916)

openvinotoolkit · Aug 30, 2024 · 5170736 · 5170736
1 parent 9265c59
commit 5170736
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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",
     },
 }