cuda

main.py

@@ -28,9 +28,12 @@ hand = MMPoseInferencer("hand")
 human3d = MMPoseInferencer(pose3d="human3d")
 track_model = YOLO('yolov8n.pt')  # Load an official Detect model
 
+
 # ultraltics
 if torch.cuda.is_available():
     device = "cuda"
+else:
+    device = "cpu"
 
 print("[INFO]: Downloaded models!")
 
@@ -53,6 +56,8 @@ def check_extension(video):
 
 def tracking(video, model, boxes=True):
     print("[INFO] Is cuda available? ", torch.cuda.is_available())
+    print(device)
+
     print("[INFO] Loading model...")
     # Load an official or custom model
 
@@ -101,6 +106,7 @@ def show_tracking(video_content):
 
 def pose3d(video):
     video = check_extension(video)
+    print(device)
 
 
     # Define new unique folder
@@ -124,6 +130,7 @@ def pose3d(video):
 
 def pose2d(video, kpt_threshold):
     video = check_extension(video)
+    print(device)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
@@ -148,7 +155,10 @@ def pose2d(video, kpt_threshold):
 
 def pose2dhand(video, kpt_threshold):
     video = check_extension(video)
-
+    print(device)
+    # ultraltics
+    if torch.cuda.is_available():
+        device = "cuda"
     # Define new unique folder
     add_dir = str(uuid.uuid4())
     vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)