new model path 3d
Browse files- 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")
|
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"])
|