Update bbox_head export format change and undo clamping of bbox_head. C…

…loses #430
bonn-activity-maps · Jan 18, 2021 · 6c22434 · 6c22434
1 parent 947f19c
commit 6c22434
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/python/logic/datasetService.py b/src/python/logic/datasetService.py
@@ -545,9 +545,10 @@ def export_dataset_PT(self, dataset):
                                             obj["bbox"] = kps
                                             del(obj["keypoints"])
                                         elif obj["type"] == "bbox_head":
-                                            kps = ptService.transform_to_XYWH(obj["keypoints"])
-                                            # Set kps outside the frame to an extreme within
-                                            kps = self.check_limits_kps(kps, width, height)
+                                            # kps = ptService.transform_to_XYWH(obj["keypoints"])
+                                            kps = obj["keypoints"]
+                                            # Do not clamp values of head_bboxes as per issue #430
+                                            # kps = self.check_limits_kps(kps, width, height)
                                             obj["bbox_head"] = kps
                                             del(obj["keypoints"])
                                         elif obj["type"] == "person":   # flatten keypoints array