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

替换权重文件

main
姜尧 2024-02-23 11:52:23 +08:00
parent eae71a2e7e
commit cb0d663726
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@ -63,7 +63,7 @@ def worker(id,source,region=None,stream=False):
else: else:
region = int(region) region = int(region)
print("start loading model...",id,source,region) print("start loading model...",id,source,region)
model = YOLO('./best.pt') model = YOLO('best.pt')
print("start loaded model!!!") print("start loaded model!!!")
balls = {} balls = {}
count = 0 count = 0

13
test.py
View File

@ -4,9 +4,18 @@ from ultralytics import YOLO
# Configure the tracking parameters and run the tracker # Configure the tracking parameters and run the tracker
# model = YOLO('best.pt') # 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" # 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 = parse.urlencode({"source": source2})
# source = "./videos/123.mp4" # print(source)
# results = model.track(source=source2, conf=0.3, iou=0.5, show=True) source = "./videos/123.mp4"
# results = model.track(source=source, conf=0.3, iou=0.5, show=True)
model = YOLO('best.pt') model = YOLO('best.pt')
results = model.track(source="./videos/123.mp4", tracker='botsort.yaml', show=True) 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