shouzhou.bx wenmeng.zwm 3 years ago
parent
commit
2223b9f16a
3 changed files with 9 additions and 6 deletions
  1. +3
    -3
      modelscope/outputs.py
  2. +4
    -1
      modelscope/pipelines/cv/body_2d_keypoints_pipeline.py
  3. +2
    -2
      modelscope/utils/cv/image_utils.py

+ 3
- 3
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:


+ 4
- 1
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
}


+ 2
- 2
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):


Loading…
Cancel
Save