From 4d3a8aa0280c0271e85c0f2127f0f6f437488df2 Mon Sep 17 00:00:00 2001 From: David PAGNON Date: Fri, 8 Nov 2024 17:37:14 +0100 Subject: [PATCH] segmentation from bbox instead of empty --- Pose2Sim/Utilities/reproj_from_trc_calib.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Pose2Sim/Utilities/reproj_from_trc_calib.py b/Pose2Sim/Utilities/reproj_from_trc_calib.py index 24304b2..67163d3 100644 --- a/Pose2Sim/Utilities/reproj_from_trc_calib.py +++ b/Pose2Sim/Utilities/reproj_from_trc_calib.py @@ -323,20 +323,20 @@ def dataset_to_mmpose2d(coords_df, mmpose_json_file, img_size, markerset='custom coords_mk = coords.loc[coords.index.get_level_values(2)==marker] coords_list += coords_mk.tolist()+[2] if not np.isnan(coords_mk).any() else [0.0, 0.0, 0] + num_keypoints = len(marker_list) + # bbox x_coords = coords.loc[coords.index.get_level_values(3)=='x'] y_coords = coords.loc[coords.index.get_level_values(3)=='y'] min_x, min_y, max_x, max_y = np.nanmin(x_coords), np.nanmin(y_coords), np.nanmax(x_coords), np.nanmax(y_coords) bbox_width = max_x - min_x bbox_height = max_y - min_y - - # num_keypoints, id, category_id - num_keypoints = len(marker_list) # bbox = [min_x, min_y, max_x, max_y] bbox = [min_x, min_y, bbox_width, bbox_height] # coco format + id = person_ids[p] category_id = 1 - segmentation = [] # no segmentation + segmentation = [[min_x, min_y, min_x, max_y, max_x, max_y, max_x, min_y]] # no segmentation area = bbox_width * bbox_height iscrowd = 0 # each annotation represents one single person