luanhailiang
/
billiard
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

app

main
luanhailiang 2023-09-22 11:27:05 +08:00
parent 7087e87559
commit b4d2cb204e
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app.py
View File

@ -72,8 +72,12 @@ def worker(id,source,region=None,stream=False):
if not cap.isOpened(): if not cap.isOpened():
print("Error opening video stream..") print("Error opening video stream..")
cap = cv2.VideoCapture(source) cap = cv2.VideoCapture(source)
time.sleep(1)
if not cap.isOpened(): if not cap.isOpened():
print("Error opening video stream...") print("Error opening video stream...")
time.sleep(3)
if not cap.isOpened():
print("Error opening video stream....")
break break
continue continue
result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%4) result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%4)
@ -81,7 +85,11 @@ def worker(id,source,region=None,stream=False):
detections = sv.Detections.from_yolov8(result) detections = sv.Detections.from_yolov8(result)
if result.boxes.id is not None: if result.boxes.id is not None:
detections.tracker_id = result.boxes.id.cpu().numpy().astype(int) detections.tracker_id = result.boxes.id.cpu().numpy().astype(int)
detections = detections[(detections.tracker_id != None)] else:
detections.tracker_id = np.array([])
detections.conf = np.array([])
detections.xyxy=np.empty((0, 4), dtype=np.float32)
# detections = detections[(detections.tracker_id != None)]
count += 1 count += 1
names = {} names = {}