make directory
Browse files
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.
|
169 |
-
video_output2 = gr.
|
170 |
-
video_output3 = gr.
|
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.
|
187 |
-
webcam_output2 = gr.
|
188 |
-
webcam_output3 = gr.
|
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 |
|