22 lines
1.0 KiB
Python
22 lines
1.0 KiB
Python
from urllib import parse
|
|
|
|
from ultralytics import YOLO
|
|
|
|
# Configure the tracking parameters and run the tracker
|
|
# model = YOLO('best1.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('best640.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
|