xmrt commited on
Commit
1253964
·
1 Parent(s): 0447c05

moved device and added mkdn

Browse files
Files changed (3) hide show
  1. cocohand.png +0 -0
  2. cocoposes.png +0 -0
  3. main_noweb.py +12 -4
cocohand.png ADDED
cocoposes.png ADDED
main_noweb.py CHANGED
@@ -95,7 +95,9 @@ def pose3d(video, kpt_threshold):
95
  video = check_extension(video)
96
  print(device)
97
 
98
- human3d = MMPoseInferencer(pose3d="human3d")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
 
 
99
 
100
  # Define new unique folder
101
  add_dir = str(uuid.uuid4())
@@ -106,11 +108,10 @@ def pose3d(video, kpt_threshold):
106
  #video = human3d.preprocess(video, batch_size=8)
107
  result_generator = human3d(video,
108
  vis_out_dir = add_dir,
109
- radius = 5,
110
- thickness=4,
111
  rebase_keypoint_height=True,
112
  kpt_thr=kpt_threshold,
113
- device=device,
114
  pred_out_dir = add_dir
115
  )
116
 
@@ -302,6 +303,13 @@ print("Angle is: ", angle)
302
  15: Left Ankle
303
  16: Right Ankle
304
  ```
 
 
 
 
 
 
 
305
  """)
306
 
307
 
 
95
  video = check_extension(video)
96
  print(device)
97
 
98
+ human3d = MMPoseInferencer(device=device,
99
+ pose3d="human3d",
100
+ scope="mmpose")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
101
 
102
  # Define new unique folder
103
  add_dir = str(uuid.uuid4())
 
108
  #video = human3d.preprocess(video, batch_size=8)
109
  result_generator = human3d(video,
110
  vis_out_dir = add_dir,
111
+ radius = 8,
112
+ thickness = 5,
113
  rebase_keypoint_height=True,
114
  kpt_thr=kpt_threshold,
 
115
  pred_out_dir = add_dir
116
  )
117
 
 
303
  15: Left Ankle
304
  16: Right Ankle
305
  ```
306
+
307
+ \n Below, you can see a visualization of the poses of the 2 and 3d keypoint locations:
308
+ <img src="./cocoposes.png" width="200">
309
+
310
+ \n And the hand keypoints are as follows:
311
+ <img src="./cocohand.png" width="200">
312
+
313
  """)
314
 
315