Add Keypoint Detection legacy template (#4094)

added rtmpose_template

Author: kprokofi

src/otx/core/model/keypoint_detection.py

@@ -11,7 +11,7 @@
 import torch
 
 from otx.algo.utils.mmengine_utils import load_checkpoint
-from otx.core.data.entity.base import OTXBatchLossEntity
+from otx.core.data.entity.base import ImageInfo, OTXBatchLossEntity
 from otx.core.data.entity.keypoint_detection import KeypointDetBatchDataEntity, KeypointDetBatchPredEntity
 from otx.core.metrics import MetricCallable, MetricInput
 from otx.core.metrics.pck import PCKMeasureCallable
@@ -150,6 +150,40 @@ def forward_for_tracing(self, image: torch.Tensor) -> torch.Tensor | tuple[torch
         """Model forward function used for the model tracing during model exportation."""
         return self.model.forward(inputs=image, mode="tensor")
 
+    def get_dummy_input(self, batch_size: int = 1) -> KeypointDetBatchDataEntity:
+        """Generates a dummy input, suitable for launching forward() on it.
+
+        Args:
+            batch_size (int, optional): number of elements in a dummy input sequence. Defaults to 1.
+
+        Returns:
+            KeypointDetBatchDataEntity: An entity containing randomly generated inference data.
+        """
+        if self.input_size is None:
+            msg = f"Input size attribute is not set for {self.__class__}"
+            raise ValueError(msg)
+
+        images = torch.rand(batch_size, 3, *self.input_size)
+        infos = []
+        for i, img in enumerate(images):
+            infos.append(
+                ImageInfo(
+                    img_idx=i,
+                    img_shape=img.shape,
+                    ori_shape=img.shape,
+                ),
+            )
+        return KeypointDetBatchDataEntity(
+            batch_size,
+            images,
+            infos,
+            bboxes=[],
+            labels=[],
+            bbox_info=[],
+            keypoints=[],
+            keypoints_visible=[],
+        )
+
     @property
     def _export_parameters(self) -> TaskLevelExportParameters:
         """Defines parameters required to export a particular model implementation."""

src/otx/tools/converter.py

@@ -184,14 +184,10 @@
         "model_name": "stfpm",
     },
     # KEYPOINT_DETECTION
-    "Custom_Keypoint_Detection_Rtmpose_T": {
+    "Keypoint_Detection_RTMPose_Tiny": {
         "task": OTXTaskType.KEYPOINT_DETECTION,
         "model_name": "rtmpose_tiny",
     },
-    "Custom_Keypoint_Detection_Rtmpose_T_Single_Obj": {
-        "task": OTXTaskType.KEYPOINT_DETECTION,
-        "model_name": "rtmpose_tiny_single_obj",
-    },
 }