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

new model path 3d

Browse files
Files changed (1) hide show
  1. main_noweb.py +4 -4
main_noweb.py CHANGED
@@ -95,7 +95,7 @@ def pose3d(video, kpt_threshold):
95
  video = check_extension(video)
96
  print(device)
97
 
98
- human3d = MMPoseInferencer("pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")# pose3d="human3d")
99
 
100
  # Define new unique folder
101
  add_dir = str(uuid.uuid4())
@@ -209,9 +209,9 @@ def UI():
209
  submit_hand_file = gr.Button("Make 2d hand estimation")
210
 
211
  with gr.Row():
212
- video_output1 = gr.PlayableVideo(label = "Estimate human 2d poses", show_label=True)
213
- video_output2 = gr.PlayableVideo(label = "Estimate human 3d poses", show_label=True)
214
- video_output3 = gr.PlayableVideo(label = "Estimate human hand poses", show_label=True)
215
 
216
  gr.Markdown("Download the .json file that contains the keypoint positions for each frame in the video.")
217
  jsonoutput = gr.File(file_types=[".json"])
 
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())
 
209
  submit_hand_file = gr.Button("Make 2d hand estimation")
210
 
211
  with gr.Row():
212
+ video_output1 = gr.PlayableVideo(label = "Estimate human 2d poses", show_label=True, height=256)
213
+ video_output2 = gr.PlayableVideo(label = "Estimate human 3d poses", show_label=True, height=256)
214
+ video_output3 = gr.PlayableVideo(label = "Estimate human hand poses", show_label=True, height=256)
215
 
216
  gr.Markdown("Download the .json file that contains the keypoint positions for each frame in the video.")
217
  jsonoutput = gr.File(file_types=[".json"])