Skip to content

Commit 5170736

Browse files
authored
Fix wrong template id mapping for anomaly (#3916)
1 parent 9265c59 commit 5170736

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/otx/tools/converter.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -122,29 +122,29 @@
122122
},
123123
# ANOMALY_CLASSIFICATION
124124
"ote_anomaly_classification_padim": {
125-
"task": OTXTaskType.SEMANTIC_SEGMENTATION,
125+
"task": OTXTaskType.ANOMALY_CLASSIFICATION,
126126
"model_name": "padim",
127127
},
128128
"ote_anomaly_classification_stfpm": {
129-
"task": OTXTaskType.SEMANTIC_SEGMENTATION,
129+
"task": OTXTaskType.ANOMALY_CLASSIFICATION,
130130
"model_name": "stfpm",
131131
},
132132
# ANOMALY_DETECTION
133133
"ote_anomaly_detection_padim": {
134-
"task": OTXTaskType.SEMANTIC_SEGMENTATION,
134+
"task": OTXTaskType.ANOMALY_DETECTION,
135135
"model_name": "padim",
136136
},
137137
"ote_anomaly_detection_stfpm": {
138-
"task": OTXTaskType.SEMANTIC_SEGMENTATION,
138+
"task": OTXTaskType.ANOMALY_DETECTION,
139139
"model_name": "stfpm",
140140
},
141141
# ANOMALY_SEGMENTATION
142142
"ote_anomaly_segmentation_padim": {
143-
"task": OTXTaskType.SEMANTIC_SEGMENTATION,
143+
"task": OTXTaskType.ANOMALY_SEGMENTATION,
144144
"model_name": "padim",
145145
},
146146
"ote_anomaly_segmentation_stfpm": {
147-
"task": OTXTaskType.SEMANTIC_SEGMENTATION,
147+
"task": OTXTaskType.ANOMALY_SEGMENTATION,
148148
"model_name": "stfpm",
149149
},
150150
}

0 commit comments

Comments
 (0)