billiard/test.py

13 lines
579 B
Python
Raw Normal View History

2023-09-27 08:05:26 +00:00
from urllib import parse
2023-08-06 05:43:37 +00:00
from ultralytics import YOLO
# Configure the tracking parameters and run the tracker
2023-10-07 05:54:35 +00:00
# model = YOLO('best.pt')
# source2 = "rtmp://rtmp03open.ys7.com:1935/v3/openlive/L05874022_1_1?expire=1726487667&id=625419662355599360&t=228dca5d37b6bead201e80cca3fc5fe74140242278e4eb51a99bcce593cae027&ev=100"
2023-09-27 08:05:26 +00:00
# source = parse.urlencode({"source": source2})
# source = "./videos/123.mp4"
2023-10-07 05:54:35 +00:00
# results = model.track(source=source2, conf=0.3, iou=0.5, show=True)
2023-08-06 05:43:37 +00:00
model = YOLO('best.pt')
2023-09-26 12:07:44 +00:00
results = model.track(source="./videos/123.mp4", tracker='botsort.yaml', show=True)