From 3efc314d53fe9cb41d5ebf263a74e7c92e642466 Mon Sep 17 00:00:00 2001 From: Vladisalv Sovrasov Date: Thu, 5 Dec 2024 23:00:09 +0900 Subject: [PATCH] Update classification in converter --- src/otx/tools/converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/tools/converter.py b/src/otx/tools/converter.py index 13edc6f1b5..81b60c9040 100644 --- a/src/otx/tools/converter.py +++ b/src/otx/tools/converter.py @@ -240,7 +240,7 @@ def convert(config_path: str, task: OTXTaskType | None = None) -> dict: if param_dict.get("enable_tiling", None) and not task_info["model_name"].endswith("_tile"): task_info["model_name"] += "_tile" # classification task type can't be deducted from template name, try to extract from config - if "sub_task_type" in template_config and "CLS" in template_config["sub_task_type"]: + if "sub_task_type" in template_config and "_CLS" in task_info["task"]: task_info["task"] = template_config["sub_task_type"] if task is not None: task_info["task"] = task