xmrt commited on
Commit
5d648b4
1 Parent(s): b8eab7d
Files changed (1) hide show
  1. main.py +14 -12
main.py CHANGED
@@ -187,24 +187,26 @@ def run_UI():
187
  video_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
188
  video_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
189
  video_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand poses", show_label=True)
190
- video_output4 = gr.Video(height=512)
191
 
192
  with gr.Tab("Record video with webcam"):
193
- with gr.Row():
194
- with gr.Column():
195
- webcam_input = gr.Video(source="webcam", height=612)
196
-
197
- web_kpthr = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Keypoint threshold')
198
-
199
- submit_pose_web = gr.Button("Make 2d pose estimation")
200
- submit_pose3d_web = gr.Button("Make 3d pose estimation")
201
- submit_hand_web = gr.Button("Make 2d hand estimation")
202
- submit_detect_web = gr.Button("Detect and track objects")
 
 
203
 
204
  webcam_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
205
  webcam_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
206
  webcam_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand position", show_label=True)
207
- webcam_output4 = gr.Video(height=512)
208
 
209
 
210
 
 
187
  video_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
188
  video_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
189
  video_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand poses", show_label=True)
190
+ video_output4 = gr.Video(height=512, label = "Detection and tracking", show_label=True, format="mp4")
191
 
192
  with gr.Tab("Record video with webcam"):
193
+
194
+ with gr.Column:
195
+ with gr.Row():
196
+ with gr.Column():
197
+ webcam_input = gr.Video(source="webcam", height=612)
198
+
199
+ web_kpthr = gr.Slider(minimum=0.1, maximum=1, step=1e3, default=0.3, label='Keypoint threshold')
200
+
201
+ submit_pose_web = gr.Button("Make 2d pose estimation")
202
+ submit_pose3d_web = gr.Button("Make 3d pose estimation")
203
+ submit_hand_web = gr.Button("Make 2d hand estimation")
204
+ submit_detect_web = gr.Button("Detect and track objects")
205
 
206
  webcam_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
207
  webcam_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
208
  webcam_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand position", show_label=True)
209
+ webcam_output4 = gr.Video(height=512, label = "Detection and tracking", show_label=True, format="mp4")
210
 
211
 
212