Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] format body 2d keypoint output boxes

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10158585
This commit is contained in:
shouzhou.bx
2022-09-19 09:44:19 +08:00
committed by wenmeng.zwm
parent d5c5c64dc8
commit df533817b8
3 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/outputs.py
View File

@@ -202,9 +202,9 @@ TASK_OUTPUTS = {
# [[score]*15]
# ]
# "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:

5
modelscope/pipelines/cv/body_2d_keypoints_pipeline.py
View File

@@ -76,8 +76,11 @@ class Body2DKeypointsPipeline(Pipeline):
}
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 {
OutputKeys.BOXES: boxes,
OutputKeys.BOXES: result_boxes,
OutputKeys.POSES: poses,
OutputKeys.SCORES: scores
}

4
modelscope/utils/cv/image_utils.py
View File

@@ -66,8 +66,8 @@ def draw_joints(image, np_kps, score, threshold=0.2):
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):