Added ultralytics

main.py

@@ -21,12 +21,12 @@ print("[INFO]: Imported modules!")
 human = MMPoseInferencer("human")
 hand = MMPoseInferencer("hand")
 human3d = MMPoseInferencer(pose3d="human3d")
+track_model = YOLO('yolov8n.pt')  # Load an official Detect model
+
 # ultraltics
 
 # Defining inferencer models to lookup in function
-inferencers = {"Estimate human 2d poses":human, "Estimate human 2d hand poses":hand, "Estimate human 3d poses":human3d}
-
-track_model = YOLO('yolov8n.pt')  # Load an official Detect model
+inferencers = {"Estimate human 2d poses":human, "Estimate human 2d hand poses":hand, "Estimate human 3d poses":human3d, "Detect and track":track_model}
 
 print("[INFO]: Downloaded models!")
 
@@ -41,13 +41,11 @@ def tracking(video, model, boxes=True):
 
     return annotated_frame
 
-
-
-def show_tracking(video_content, vis_out_dir):
+def show_tracking(video_content, vis_out_dir, model):
         video = cv2.VideoCapture(video_content)
 
         # Track
-        video_track = tracking(video_content, track_model.track)
+        video_track = tracking(video_content, model                                                                                 .track)
 
         # Prepare to save video
         out_file = os.path.join(vis_out_dir, "track.mp4")
@@ -97,7 +95,8 @@ def infer(video, check):
         inferencer = inferencers[i] # 'hand', 'human , device='cuda'
 
         print("[INFO]: Running inference!")
-        out_file = poses(inferencer, video, vis_out_dir)
+        out_file = poses(inferencer, video)                                                                                                                                                                                                    o, vis_out_dir)
+
         if i == "Detect and track":
             show_tracking(video, vis_out_dir)