部分小问题的修改

apps/business/deepsort/service.py

@@ -3,12 +3,12 @@ from utils.websocket_server import room_manager
 
 import time
 import torch
+from deep_sort.deep_sort import DeepSort
+from deep_sort.utils.draw import draw_boxes
 from utils.yolov5.models.common import DetectMultiBackend
 from utils.yolov5.utils.torch_utils import select_device
 from utils.yolov5.utils.dataloaders import LoadImages, LoadStreams
 from utils.yolov5.utils.general import check_img_size, non_max_suppression, cv2, scale_coords, xyxy2xywh
-from deep_sort.deep_sort import DeepSort
-from deep_sort.utils.draw import draw_boxes
 
 
 def yolov5_to_deepsort_format(pred):
@@ -67,9 +67,16 @@ async def run_deepsort(
 
     model.warmup(imgsz=(1 if pt or model.triton else bs, 3, *img_sz))
 
-    time.sleep(2)  # 等待2s，等待websocket进入
+    time.sleep(3)  # 等待3s，等待websocket进入
+
+    start_time = time.time()
 
     for path, im, im0s, vid_cap, s in dataset:
+        # 检查是否已经超过10分钟（600秒）
+        elapsed_time = time.time() - start_time
+        if elapsed_time > 600:  # 600 seconds = 10 minutes
+            print(room, "已达到最大执行时间，结束推理。")
+            break
         if room_manager.rooms.get(room):
             im0 = im0s[0]
             im = torch.from_numpy(im).to(model.device)