Detect num_classes information for the H-label classification (#3064)

* Initial commit

* Add the logic for auto hlabel information

* Fix precommit

src/otx/cli/cli.py

@@ -422,6 +422,14 @@ def instantiate_model(self, model_config: Namespace) -> tuple:
                 warn(warning_msg, stacklevel=0)
                 model_config.init_args.num_classes = num_classes
 
+            # Hlabel classification
+            from otx.core.data.dataset.classification import HLabelInfo
+
+            if isinstance(self.datamodule.label_info, HLabelInfo):
+                hlabel_info = self.datamodule.label_info
+                model_config.init_args.num_multiclass_heads = hlabel_info.num_multiclass_heads
+                model_config.init_args.num_multilabel_classes = hlabel_info.num_multilabel_classes
+
         # Parses the OTXModel separately to update num_classes.
         model_parser = ArgumentParser()
         model_parser.add_subclass_arguments(OTXModel, "model", required=False, fail_untyped=False)

src/otx/engine/utils/auto_configurator.py

@@ -257,6 +257,14 @@ def get_model(self, model_name: str | None = None, label_info: LabelInfo | None
         if label_info is not None:
             num_classes = label_info.num_classes
             self.config["model"]["init_args"]["num_classes"] = num_classes
+
+            from otx.core.data.dataset.classification import HLabelInfo
+
+            if isinstance(label_info, HLabelInfo):
+                init_args = self.config["model"]["init_args"]
+                init_args["num_multiclass_heads"] = label_info.num_multiclass_heads
+                init_args["num_multilabel_classes"] = label_info.num_multilabel_classes
+
         logger.warning(f"Set Default Model: {self.config['model']}")
         return instantiate_class(args=(), init=self.config["model"])
 

tests/integration/cli/test_cli.py

@@ -62,10 +62,6 @@ def test_otx_e2e(
         "1" if task in ("zero_shot_visual_prompting") else "2",
         *fxt_cli_override_command_per_task[task],
     ]
-    # H-Label-CLS need to add --metric
-    if task in ("h_label_cls"):
-        command_cfg.extend(["--metric.num_multiclass_heads", "2"])
-        command_cfg.extend(["--metric.num_multilabel_classes", "3"])
 
     run_main(command_cfg=command_cfg, open_subprocess=fxt_open_subprocess)
 
@@ -105,10 +101,6 @@ def test_otx_e2e(
         "--checkpoint",
         str(ckpt_files[-1]),
     ]
-    # H-Label-CLS need to add --metric
-    if task in ("h_label_cls"):
-        command_cfg.extend(["--metric.num_multiclass_heads", "2"])
-        command_cfg.extend(["--metric.num_multilabel_classes", "3"])
 
     run_main(command_cfg=command_cfg, open_subprocess=fxt_open_subprocess)
 

tests/integration/conftest.py

@@ -145,12 +145,7 @@ def fxt_cli_override_command_per_task() -> dict:
     return {
         "multi_class_cls": [],
         "multi_label_cls": [],
-        "h_label_cls": [
-            "--model.num_multiclass_heads",
-            "2",
-            "--model.num_multilabel_classes",
-            "3",
-        ],
+        "h_label_cls": [],
         "detection": [],
         "rotated_detection": [],
         "instance_segmentation": [],