Update keypoint detection export with explain (#4192)

* Revert "Fix KP Export (single obj case) (#4191)"

This reverts commit 35fdb7f.

* Update aouto configuration to support kp format

* Relax KP explain export checks

Author: sovrasov

src/otx/algo/keypoint_detection/rtmpose.py

@@ -5,6 +5,7 @@
 
 from __future__ import annotations
 
+import logging
 from typing import TYPE_CHECKING
 
 from otx.algo.common.backbones import CSPNeXt
@@ -25,6 +26,9 @@
     from otx.core.types.label import LabelInfoTypes
 
 
+logger = logging.getLogger()
+
+
 class RTMPose(OTXKeypointDetectionModel):
     """OTX keypoint detection model class for RTMPose."""
 
@@ -37,7 +41,7 @@ def _exporter(self) -> OTXModelExporter:
 
         if self.explain_mode:
             msg = "Export with explain is not supported for RTMPose model."
-            raise ValueError(msg)
+            logger.warning(msg)
 
         return OTXNativeModelExporter(
             task_level_export_parameters=self._export_parameters,

src/otx/engine/utils/auto_configurator.py

@@ -55,6 +55,7 @@
 TASK_PER_DATA_FORMAT = {
     "imagenet_with_subset_dirs": [OTXTaskType.MULTI_CLASS_CLS, OTXTaskType.H_LABEL_CLS],
     "datumaro": [OTXTaskType.MULTI_LABEL_CLS],
+    "coco_person_keypoints": [OTXTaskType.KEYPOINT_DETECTION],
     "coco_instances": [
         OTXTaskType.DETECTION,
         OTXTaskType.ROTATED_DETECTION,

src/otx/recipe/keypoint_detection/openvino_model.yaml

@@ -26,6 +26,7 @@ overrides:
       batch_size: 1
       num_workers: 2
       transforms:
+        - class_path: otx.core.data.transform_libs.torchvision.GetBBoxCenterScale
         - class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
           init_args:
             input_size: $(input_size)
@@ -34,6 +35,7 @@ overrides:
       batch_size: 1
       num_workers: 2
       transforms:
+        - class_path: otx.core.data.transform_libs.torchvision.GetBBoxCenterScale
         - class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
           init_args:
             input_size: $(input_size)
@@ -42,6 +44,7 @@ overrides:
       batch_size: 64
       num_workers: 2
       transforms:
+        - class_path: otx.core.data.transform_libs.torchvision.GetBBoxCenterScale
         - class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
           init_args:
             input_size: $(input_size)