Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10158585master
| @@ -202,9 +202,9 @@ TASK_OUTPUTS = { | |||||
| # [[score]*15] | # [[score]*15] | ||||
| # ] | # ] | ||||
| # "boxes": [ | # "boxes": [ | ||||
| # [[x1, y1], [x2, y2]], | |||||
| # [[x1, y1], [x2, y2]], | |||||
| # [[x1, y1], [x2, y2]], | |||||
| # [x1, y1, x2, y2], | |||||
| # [x1, y1, x2, y2], | |||||
| # [x1, y1, x2, y2], | |||||
| # ] | # ] | ||||
| # } | # } | ||||
| Tasks.body_2d_keypoints: | Tasks.body_2d_keypoints: | ||||
| @@ -76,8 +76,11 @@ class Body2DKeypointsPipeline(Pipeline): | |||||
| } | } | ||||
| poses, scores, boxes = self.keypoint_model.postprocess(input) | poses, scores, boxes = self.keypoint_model.postprocess(input) | ||||
| result_boxes = [] | |||||
| for box in boxes: | |||||
| result_boxes.append([box[0][0], box[0][1], box[1][0], box[1][1]]) | |||||
| return { | return { | ||||
| OutputKeys.BOXES: boxes, | |||||
| OutputKeys.BOXES: result_boxes, | |||||
| OutputKeys.POSES: poses, | OutputKeys.POSES: poses, | ||||
| OutputKeys.SCORES: scores | OutputKeys.SCORES: scores | ||||
| } | } | ||||
| @@ -66,8 +66,8 @@ def draw_joints(image, np_kps, score, threshold=0.2): | |||||
| def draw_box(image, box): | def draw_box(image, box): | ||||
| cv2.rectangle(image, (int(box[0][0]), int(box[0][1])), | |||||
| (int(box[1][0]), int(box[1][1])), (0, 0, 255), 2) | |||||
| cv2.rectangle(image, (int(box[0]), int(box[1])), | |||||
| (int(box[2]), int(box[3])), (0, 0, 255), 2) | |||||
| def realtime_object_detection_bbox_vis(image, bboxes): | def realtime_object_detection_bbox_vis(image, bboxes): | ||||