from urllib import parse

from ultralytics import YOLO

# Configure the tracking parameters and run the tracker
# model = YOLO('best.pt')
# source2 = "rtmp://rtmp01open.ys7.com:1935/v3/openlive/BB2712256_1_2?expire=1733477171&id=654735767607717888&t=a28eb41ed3e1700eb603220184b1a1776af461239b8c483f9f554adb0ed36df6&ev=100"
# source2 = parse.urlencode({"source": source_single_url})
# source2 = "rtmp://rtmp03open.ys7.com:1935/v3/openlive/L05874022_1_1?expire=1726487667&id=625419662355599360&t=228dca5d37b6bead201e80cca3fc5fe74140242278e4eb51a99bcce593cae027&ev=100"
# source = parse.urlencode({"source": source2})
# print(source)
source = "./videos/123.mp4"
# results = model.track(source=source, conf=0.3, iou=0.5, show=True)
model = YOLO('best.pt')
results = model.track(source="./videos/123.mp4", tracker='botsort.yaml', show=True)
# results = model(source=source2, tracker='botsort.yaml', show=True, stream=True)
for result in results:
    boxes = result.boxes
    masks = result.masks
    keypoints = result.keypoints
    probs = result.probs