xmrt commited on
Commit
8ba8f27
1 Parent(s): ae92d0e

make directory

Browse files
Files changed (1) hide show
  1. main.py +11 -9
main.py CHANGED
@@ -142,10 +142,12 @@ def pose2dhand(video):
142
  device="cuda")
143
 
144
  result = [result for result in result_generator] #next(result_generator)
145
-
 
 
146
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
147
- print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
148
-
149
  return out_file
150
 
151
  if __name__ == "__main__":
@@ -165,9 +167,9 @@ if __name__ == "__main__":
165
  submit_hand_file = gr.Button("Make 2d hand estimation")
166
  submit_detect_file = gr.Button("Detect and track objects")
167
 
168
- video_output1 = gr.Video(height=512)
169
- video_output2 = gr.Video(height=512)
170
- video_output3 = gr.Video(height=512)
171
  video_output4 = gr.Video(height=512)
172
 
173
  with gr.Tab("Record video with webcam"):
@@ -183,9 +185,9 @@ if __name__ == "__main__":
183
  submit_hand_web = gr.Button("Make 2d hand estimation")
184
  submit_detect_web = gr.Button("Detect and track objects")
185
 
186
- webcam_output1 = gr.Video(height=512)
187
- webcam_output2 = gr.Video(height=512)
188
- webcam_output3 = gr.Video(height=512)
189
  webcam_output4 = gr.Video(height=512)
190
 
191
 
 
142
  device="cuda")
143
 
144
  result = [result for result in result_generator] #next(result_generator)
145
+
146
+ print(glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm")))
147
+
148
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm"))
149
+ #print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
150
+ out_file = video
151
  return out_file
152
 
153
  if __name__ == "__main__":
 
167
  submit_hand_file = gr.Button("Make 2d hand estimation")
168
  submit_detect_file = gr.Button("Detect and track objects")
169
 
170
+ video_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
171
+ video_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
172
+ video_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand poses", show_label=True)
173
  video_output4 = gr.Video(height=512)
174
 
175
  with gr.Tab("Record video with webcam"):
 
185
  submit_hand_web = gr.Button("Make 2d hand estimation")
186
  submit_detect_web = gr.Button("Detect and track objects")
187
 
188
+ webcam_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
189
+ webcam_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
190
+ webcam_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand position", show_label=True)
191
  webcam_output4 = gr.Video(height=512)
192
 
193