glob

main.py

@@ -92,8 +92,9 @@ def pose3d(video):
                                  rebase_keypoint_height=True,
                                  device="cuda")    
     
+    print("OUTDIR", os.listdir(os.path.join(vis_out_dir)))
+
     result = [result for result in result_generator] #next(result_generator)    
-    out_file = glob.glob(os.path.join(vis_out_dir, "*"))
     out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) + glob.glob(os.path.join(vis_out_dir, "*.webm")) 
     print("[INFO]: CURRENT OUT FILE NAME: ", out_file)
 
@@ -108,10 +109,12 @@ def pose2d(video):
 
     result_generator = human(video, 
                                  vis_out_dir = vis_out_dir,
-                                 thickness=2,
                                  return_vis=True,
+                                 thickness=2,
                                  rebase_keypoint_height=True,
-                                 device="cuda")    
+                                 #kpt_thr=kpt_thr,
+                                 device="cuda"
+                                 )    
     
     result = [result for result in result_generator] #next(result_generator)    
 
@@ -145,78 +148,79 @@ def pose2dhand(video):
 
     return out_file
 
-
-
-with gr.Blocks() as demo:
-    with gr.Column():            
-        with gr.Tab("Upload video"):
-            with gr.Row():
-                with gr.Column():
-                    video_input = gr.Video(source="upload", type="filepath", height=512)
-                    # Insert slider with kpt_thr
-                    file_kpthr = gr.Slider(0, 1, value=0.3)
-
-                    submit_pose_file = gr.Button("Make 2d pose estimation")
-                    submit_pose3d_file = gr.Button("Make 3d pose estimation")
-                    submit_hand_file = gr.Button("Make 2d hand estimation")
-                    submit_detect_file = gr.Button("Detect and track objects")
-
-                video_output1 = gr.Video(height=512)
-                video_output2 = gr.Video(height=512)
-                video_output3 = gr.Video(height=512)
-                video_output4 = gr.Video(height=512)
-
-        with gr.Tab("Record video with webcam"):
-            with gr.Row():
-                with gr.Column():
-                    webcam_input = gr.Video(source="webcam", height=512)
-                    
-                    web_kpthr = gr.Slider(0, 1, value=0.3)
-
-                    submit_pose_web = gr.Button("Make 2d pose estimation")
-                    submit_pose3d_web = gr.Button("Make 3d pose estimation")
-                    submit_hand_web = gr.Button("Make 2d hand estimation")
-                    submit_detect_web = gr.Button("Detect and track objects")
-
-                webcam_output1 = gr.Video(height=512)
-                webcam_output2 = gr.Video(height=512)
-                webcam_output3 = gr.Video(height=512)
-                webcam_output4 = gr.Video(height=512)
-
+if __name__ == "__main__":
+
+
+    with gr.Blocks() as demo:
+        with gr.Column():            
+            with gr.Tab("Upload video"):
+                with gr.Row():
+                    with gr.Column():
+                        video_input = gr.Video(source="upload", type="filepath", height=512)
+                        # Insert slider with kpt_thr
+                        file_kpthr = gr.Slider(0, 1, value=0.3)
+
+                        submit_pose_file = gr.Button("Make 2d pose estimation")
+                        submit_pose3d_file = gr.Button("Make 3d pose estimation")
+                        submit_hand_file = gr.Button("Make 2d hand estimation")
+                        submit_detect_file = gr.Button("Detect and track objects")
+
+                    video_output1 = gr.Video(height=512)
+                    video_output2 = gr.Video(height=512)
+                    video_output3 = gr.Video(height=512)
+                    video_output4 = gr.Video(height=512)
+
+            with gr.Tab("Record video with webcam"):
+                with gr.Row():
+                    with gr.Column():
+                        webcam_input = gr.Video(source="webcam", height=512)
+                        
+                        web_kpthr = gr.Slider(0, 1, value=0.3)
+
+                        submit_pose_web = gr.Button("Make 2d pose estimation")
+                        submit_pose3d_web = gr.Button("Make 3d pose estimation")
+                        submit_hand_web = gr.Button("Make 2d hand estimation")
+                        submit_detect_web = gr.Button("Detect and track objects")
+
+                    webcam_output1 = gr.Video(height=512)
+                    webcam_output2 = gr.Video(height=512)
+                    webcam_output3 = gr.Video(height=512)
+                    webcam_output4 = gr.Video(height=512)
+
+            
         
-    
-    # From file
-    submit_pose_file.click(fn=pose2d, 
-                           inputs= video_input, 
-                           outputs = video_output1)
-    
-    submit_pose3d_file.click(fn=pose3d, 
-                             inputs= video_input, 
-                             outputs = video_output2)
-    
-    submit_hand_file.click(fn=pose2dhand, 
-                           inputs= video_input, 
-                           outputs = video_output3)
-    
-    submit_detect_file.click(fn=show_tracking, 
-                             inputs= video_input, 
-                             outputs = video_output4)
-    
-    # Web
-    submit_pose_web.click(fn=pose2d, 
-                          inputs= webcam_input, 
-                          outputs = webcam_output1)
-    
-    submit_pose3d_web.click(fn=pose3d, 
-                            inputs= webcam_input, 
-                            outputs = webcam_output2)
-    
-    submit_hand_web.click(fn=pose2dhand, 
-                          inputs= webcam_input, 
-                          outputs = video_output3)
-    
-    submit_detect_web.click(fn=show_tracking, 
+        # From file
+        submit_pose_file.click(fn=pose2d, 
+                            inputs= video_input, 
+                            outputs = video_output1)
+        
+        submit_pose3d_file.click(fn=pose3d, 
+                                inputs= video_input, 
+                                outputs = video_output2)
+        
+        submit_hand_file.click(fn=pose2dhand, 
+                            inputs= video_input, 
+                            outputs = video_output3)
+        
+        submit_detect_file.click(fn=show_tracking, 
+                                inputs= video_input, 
+                                outputs = video_output4)
+        
+        # Web
+        submit_pose_web.click(fn=pose2d, 
+                            inputs = webcam_input, 
+                            outputs = webcam_output1)
+        
+        submit_pose3d_web.click(fn=pose3d, 
+                                inputs= webcam_input, 
+                                outputs = webcam_output2)
+        
+        submit_hand_web.click(fn=pose2dhand, 
                             inputs= webcam_input, 
-                            outputs = video_output4)
+                            outputs = video_output3)
+        
+        submit_detect_web.click(fn=show_tracking, 
+                                inputs= webcam_input, 
+                                outputs = video_output4)
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)