diff --git a/app.py b/app.py index c6f40b7..910b397 100644 --- a/app.py +++ b/app.py @@ -68,7 +68,11 @@ def worker(id,source,region=None,stream=False): ret, frame = cap.read() if not ret: print("Error read video stream.") - break + cap = cv2.VideoCapture(source) + if not cap.isOpened(): + print("Error opening video stream.") + break + continue result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%4) result = result[0] detections = sv.Detections.from_yolov8(result)