add bbox info

tools/inference_custom.py

@@ -26,6 +26,13 @@ def parse_args():
     return parser.parse_args()
 
 
+def draw_bboxes(image, bboxes):
+    """Draw bounding boxes on the image using OpenCV."""
+    for bbox in bboxes:
+        x, y, x2, y2 = map(int, bbox)
+        cv2.rectangle(image, (x, y), (x2, y2), (255, 0, 0), 2)  # Draw rectangle
+
+
 def draw_keypoints(image, keypoints, scores, score_thr):
     """Draw keypoints on the image using OpenCV."""
     for el1, el2 in zip(keypoints, scores):
@@ -78,7 +85,7 @@ def main():
             model,
             image_rgb,
             person_bboxes,
-            bbox_format='xywh'  # COCO annotations typically use 'xywh' format
+            bbox_format='xyxy'  # COCO annotations typically use 'xywh' format
         )
 
         keypoints_results = []
@@ -87,19 +94,25 @@ def main():
             if pred_instances is not None:
                 keypoints = pred_instances.keypoints
                 scores = pred_instances.keypoint_scores
+                bbox = person_bboxes[0]  # Taking first bbox, adjust for multiple detections
+
+                # Save the bbox along with the keypoints data
                 keypoints_results.append({
                     'keypoints': keypoints.tolist(),
-                    'scores': scores.tolist()
+                    'scores': scores.tolist(),
+                    'bbox': bbox.tolist()  # Add bbox to the result
                 })
 
                 if args.out_dir:
                     draw_keypoints(image, keypoints, scores, args.score_thr)
+                    draw_bboxes(image, person_bboxes)
 
         # Save the visualized image if `out-dir` is provided
         if args.out_dir:
             out_file = os.path.join(args.out_dir, img_info['file_name'])
             cv2.imwrite(out_file, image)
 
+        # Save individual prediction file
         prediction_file = os.path.join(args.predictions_dir, f"{os.path.splitext(img_info['file_name'])[0]}.json")
         with open(prediction_file, 'w') as f:
             json.dump({