xmrt commited on
Commit
b705b5d
·
1 Parent(s): fe44aa1
Files changed (1) hide show
  1. main_noweb.py +5 -3
main_noweb.py CHANGED
@@ -96,9 +96,11 @@ def pose3d(video, kpt_threshold):
96
 
97
  #human3d = MMPoseInferencer(device=device, pose3d="human3d", scope="mmpose")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
98
 
99
- human3d = gr.State(value=MMPoseInferencer(device=device,
100
  pose3d="human3d",
101
- scope="mmpose"))
 
 
102
 
103
  # Define new unique folder
104
  add_dir = str(uuid.uuid4())
@@ -115,7 +117,7 @@ def pose3d(video, kpt_threshold):
115
  kpt_thr=kpt_threshold,
116
  pred_out_dir = add_dir
117
  )
118
-
119
  result = [result for result in result_generator] #next(result_generator)
120
 
121
  out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
 
96
 
97
  #human3d = MMPoseInferencer(device=device, pose3d="human3d", scope="mmpose")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
98
 
99
+ human3d = MMPoseInferencer(device=device,
100
  pose3d="human3d",
101
+ scope="mmpose")
102
+ print("HUMAN 3d downloaded!!")
103
+ #gr.State(value=)
104
 
105
  # Define new unique folder
106
  add_dir = str(uuid.uuid4())
 
117
  kpt_thr=kpt_threshold,
118
  pred_out_dir = add_dir
119
  )
120
+ print("INFERENCE DONW")
121
  result = [result for result in result_generator] #next(result_generator)
122
 
123
  out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))