Fix KP task to work in Geti scenario (#4189)

* fix KP task to work in Geti * minor
openvinotoolkit · Jan 28, 2025 · 40b82f6 · 40b82f6
1 parent 0c624a4
commit 40b82f6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/otx/core/data/dataset/keypoint_detection.py b/src/otx/core/data/dataset/keypoint_detection.py
@@ -81,9 +81,12 @@ def _get_single_bbox_dataset(self, dm_subset: DatasetSubset) -> Dataset:
                     if isinstance(ann, Points) and max(ann.points) <= 0:
                         continue
                     available_types.append(ann.type)
-                if available_types != [AnnotationType.points, AnnotationType.bbox]:
+                if AnnotationType.points not in available_types:
                     continue
                 dm_items.append(item.wrap(id=item.id + "_" + str(ann_id), annotations=anns))
+        if len(dm_items) == 0:
+            msg = "No keypoints found in the dataset. Please, check dataset annotations."
+            raise ValueError(msg)
         return Dataset.from_iterable(dm_items, categories=self.dm_subset.categories())
 
     def _get_item_impl(self, index: int) -> KeypointDetDataEntity | None:

diff --git a/src/otx/tools/converter.py b/src/otx/tools/converter.py
@@ -199,7 +199,7 @@
     # KEYPOINT_DETECTION
     "Keypoint_Detection_RTMPose_Tiny": {
         "task": OTXTaskType.KEYPOINT_DETECTION,
-        "model_name": "rtmpose_tiny",
+        "model_name": "rtmpose_tiny_single_obj",
     },
 }