new model path 3d

main_noweb.py

@@ -95,7 +95,7 @@ def pose3d(video, kpt_threshold):
     video = check_extension(video)
     print(device)
 
-    human3d = MMPoseInferencer(pose3d="human3d")
+    human3d = MMPoseInferencer("pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")#	pose3d="human3d")
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
@@ -198,7 +198,7 @@ def UI():
                     with gr.Row():
                         with gr.Column():
                             with gr.Row():
-                                video_input = gr.Video(source="upload", type="filepath", height=512, width=512)
+                                video_input = gr.Video(source="upload", type="filepath", height=256, width=192)
                                 # Insert slider with kpt_thr
                                 with gr.Column():
                                     gr.Markdown("Drag the keypoint threshold to filter out lower probability keypoints:")
@@ -209,9 +209,9 @@ def UI():
                                 submit_hand_file = gr.Button("Make 2d hand estimation")
 
                     with gr.Row():
-                        video_output1 = gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True)
-                        video_output2 = gr.PlayableVideo(height=512,  label = "Estimate human 3d poses", show_label=True)
-                        video_output3 = gr.PlayableVideo(height=512,  label = "Estimate human hand poses", show_label=True)
+                        video_output1 = gr.PlayableVideo(label = "Estimate human 2d poses", show_label=True)
+                        video_output2 = gr.PlayableVideo(label = "Estimate human 3d poses", show_label=True)
+                        video_output3 = gr.PlayableVideo(label = "Estimate human hand poses", show_label=True)
                     
                     gr.Markdown("Download the .json file that contains the keypoint positions for each frame in the video.")
                     jsonoutput = gr.File(file_types=[".json"])