billiard/test.py

13 lines
579 B
Python

from urllib import parse
from ultralytics import YOLO
# Configure the tracking parameters and run the tracker
# model = YOLO('best.pt')
# source2 = "rtmp://rtmp03open.ys7.com:1935/v3/openlive/L05874022_1_1?expire=1726487667&id=625419662355599360&t=228dca5d37b6bead201e80cca3fc5fe74140242278e4eb51a99bcce593cae027&ev=100"
# source = parse.urlencode({"source": source2})
# source = "./videos/123.mp4"
# results = model.track(source=source2, conf=0.3, iou=0.5, show=True)
model = YOLO('best.pt')
results = model.track(source="./videos/123.mp4", tracker='botsort.yaml', show=True)