check webcam

main.py

@@ -26,6 +26,8 @@ track_model = YOLO('yolov8n.pt')  # Load an official Detect model
 
 # ultraltics
 
+# [INFO] VIDEO INPUT:  /tmp/gradio/927601b660ec45919366ce37df1ed004a1fcffab/sample_flip.webm
+
 # Defining inferencer models to lookup in function
 inferencers = {"Estimate human 2d poses":human, "Estimate human 2d hand poses":hand, "Estimate human 3d poses":human3d, "Detect and track":track_model}
 
@@ -94,7 +96,7 @@ def poses(inferencer, video, vis_out_dir, kpt_thr):
 
     return out_file
 
-def infer(video, check, kpt_thr):
+def infer(video, check, kpt_thr, webcam=False):
     print("[INFO] VIDEO INPUT: ", video)
 
     # Selecting the specific inferencer
@@ -108,11 +110,14 @@ def infer(video, check, kpt_thr):
         if i == "Detect and track":
             #continue
             trackfile = show_tracking(video, vis_out_dir, inferencer)
-
+           
         else:
             out_file = poses(inferencer, video, vis_out_dir, kpt_thr)                                                                                                                                                                                                    
-        
-            out_files.extend(out_file)
+
+            if webcam==True:
+                out_files.extend(video)
+            else:
+                out_files.extend(out_file)
         
         print(out_files)
 
@@ -123,16 +128,20 @@ def run():
     check_web =  gr.CheckboxGroup(choices = ["Detect and track", "Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"], label="Methods", type="value", info="Select the model(s) you want")
     check_file = gr.CheckboxGroup(choices = ["Detect and track", "Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"], label="Methods", type="value", info="Select the model(s) you want")
     
+    description = """
+    \n\nHere you can upload videos or record one with your webcam and track objects or detect bodyposes in 2d and 3d. 
+    """
+
     # Insert slider with kpt_thr
     web_kpthr = gr.Slider(0, 1, value=0.3)
     file_kpthr = gr.Slider(0, 1, value=0.3)
     
     webcam = gr.Interface(
         fn=infer,
-        inputs= [gr.Video(source="webcam", height=512), check_web, web_kpthr],
+        inputs= [gr.Video(source="webcam", height=512), check_web, web_kpthr, True], # /tmp/gradio/927601b660ec45919366ce37df1ed004a1fcffab/sample_flip.webm
         outputs = [gr.Video(format='mp4', height=512, label="Detect and track", show_label=True), gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 2d hand poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)],
         title = 'Tracking and pose estimation', 
-        description = 'Tracking and pose estimation',
+        description = description,
         allow_flagging=False
         )
 
@@ -141,7 +150,7 @@ def run():
         inputs = [gr.Video(source="upload", height=512), check_file, file_kpthr],
         outputs = [gr.Video(format='mp4', height=512, label="Detect and track", show_label=True), gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 2d hand poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)],
         title = 'Tracking and pose estimation', 
-        description = 'Tracking and pose estimation',
+        description = description,
         allow_flagging=False
     )
 
@@ -154,9 +163,6 @@ def run():
 
 
 if __name__ == "__main__":
-    description = """
-    \n\nHere you can upload videos or record one with your webcam and track objects or detect bodyposes in 2d and 3d. 
-    """
     run()
 
 # https://github.com/open-mmlab/mmpose/tree/dev-1.x/configs/body_3d_keypoint/pose_lift