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

ball

main
luanhailiang 2023-11-07 16:16:15 +08:00
parent ef5b8df9b4
commit bd8f4a7cdd
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app.py
View File

@ -122,15 +122,15 @@ def worker(id,source,region=None,stream=False):
count += 1 count += 1
names = {} names = {}
# 检查 "xyxy" 键是否为 None
if ball.get("xyxy") is None:
ball["xyxy"] = []
for xyxy,_, confidence, class_id,tracker_id in detections: for xyxy,_, confidence, class_id,tracker_id in detections:
name = model.model.names[class_id] name = model.model.names[class_id]
if name in names: if name in names:
print("Name duplicate",name) print("Name duplicate",name)
continue continue
ball = balls.get(name,{}) ball = balls.get(name,{})
# 检查 "xyxy" 键是否为 None
if ball.get("xyxy") is None:
ball["xyxy"] = []
# ball["tkid"] = int(tracker_id) # ball["tkid"] = int(tracker_id)
ball["conf"] = round(float(confidence), 2) ball["conf"] = round(float(confidence), 2)
site = [int(xyxy[0]), int(xyxy[1]), int(xyxy[2]), int(xyxy[3])] site = [int(xyxy[0]), int(xyxy[1]), int(xyxy[2]), int(xyxy[3])]