nowweb

main.py

@@ -106,17 +106,18 @@ def show_tracking(video_content):
 
 
 def pose3d(video):
+    
     video = check_extension(video)
     print(device)
 
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
-    vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
-    os.makedirs(vis_out_dir)
+    #vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
+    os.makedirs(add_dir)
 
     result_generator = human3d(video, 
-                                vis_out_dir = vis_out_dir,
+                                vis_out_dir = add_dir,
                                 thickness=4,
                                 radius = 5,
                                 return_vis=True,
@@ -126,7 +127,7 @@ def pose3d(video):
     
     result = [result for result in result_generator] #next(result_generator)        
 
-    out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
+    out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
     
     # Reinitialize
     human3d().init()
@@ -134,29 +135,31 @@ def pose3d(video):
 
 
 def pose2d(video, kpt_threshold):
+    human = MMPoseInferencer("human")
+
     video = check_extension(video)
     print(device)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
-    vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
-    os.makedirs(vis_out_dir)
+    #vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
+    os.makedirs(add_dir)
 
     result_generator = human(video, 
-                            vis_out_dir = vis_out_dir,
+                            vis_out_dir = add_dir,
                             return_vis=True,
                             radius = 5,
                             thickness=4,
                             rebase_keypoint_height=True,
                             kpt_thr=kpt_threshold,
                             device=device,
-                            pred_out_dir = vis_out_dir
+                            pred_out_dir = add_dir
                             )    
     
     result = [result for result in result_generator] #next(result_generator)        
 
-    out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
-    kpoints = glob.glob(os.path.join(vis_out_dir, "*.json"))
+    out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
+    kpoints = glob.glob(os.path.join(add_dir, "*.json"))
 
     return "".join(out_file), "".join(kpoints)
 
@@ -168,11 +171,11 @@ def pose2dhand(video, kpt_threshold):
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
-    vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
-    os.makedirs(vis_out_dir)
+    #vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
+    os.makedirs(add_dir)
 
     result_generator = hand(video, 
-                                 vis_out_dir = vis_out_dir,
+                                 vis_out_dir = add_dir,
                                  return_vis=True,
                                  thickness = 4,
                                  radius = 5,
@@ -182,7 +185,7 @@ def pose2dhand(video, kpt_threshold):
     
     result = [result for result in result_generator] #next(result_generator)        
 
-    out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
+    out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
    
     return "".join(out_file)
 

main_noweb.py

@@ -35,7 +35,6 @@ hand = MMPoseInferencer("hand")
 human3d = MMPoseInferencer(pose3d="human3d")
 track_model = YOLO('yolov8n.pt')  # Load an official Detect model
 
-
 print("[INFO]: Downloaded models!")
 
 def check_extension(video):
@@ -139,7 +138,7 @@ def pose2d(video, kpt_threshold):
     # Define new unique folder
     add_dir = str(uuid.uuid4())
     vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
-    
+
     os.makedirs(add_dir)
 
     result_generator = human(video, 
@@ -265,7 +264,8 @@ with block:
                                 queue=False)
         
 if __name__ == "__main__":
-    block.queue(concurrency_count=10, # When you increase the concurrency_count parameter in queue(), max_threads() in launch() is automatically increased as well.
+    block.queue(share=True,
+                concurrency_count=10, # When you increase the concurrency_count parameter in queue(), max_threads() in launch() is automatically increased as well.
                 max_size=25, # Maximum number of requests that the queue processes
                 api_open = False # When creating a Gradio demo, you may want to restrict all traffic to happen through the user interface as opposed to the programmatic API that is automatically created for your Gradio demo.  
                     )  # https://www.gradio.app/guides/setting-up-a-demo-for-maximum-performance