Skip to content

Commit

Permalink
fix a bug in demo (#1373)
Browse files Browse the repository at this point in the history
  • Loading branch information
ly015 authored May 16, 2022
1 parent 04e57f8 commit f8b79fe
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions demo/body3d_two_stage_img_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,8 @@
from mmpose.apis import (inference_pose_lifter_model,
inference_top_down_pose_model, vis_3d_pose_result)
from mmpose.apis.inference import init_pose_model
from mmpose.core import SimpleCamera
from mmpose.core.bbox import bbox_xywh2xyxy
from mmpose.core.camera import SimpleCamera
from mmpose.datasets import DatasetInfo


Expand Down Expand Up @@ -127,7 +128,6 @@ def main():
# First stage: 2D pose detection
pose_det_results_list = []
if args.only_second_stage:
from mmpose.apis.inference import _xywh2xyxy

print('Stage 1: load 2D pose results from Json file.')
for image_id, image in coco.imgs.items():
Expand All @@ -144,7 +144,7 @@ def main():

pose_det_result = {
'image_name': image_name,
'bbox': _xywh2xyxy(bbox),
'bbox': bbox_xywh2xyxy(bbox),
'keypoints': keypoints,
'keypoints_3d': keypoints_3d
}
Expand Down

0 comments on commit f8b79fe

Please sign in to comment.