系统配置问题

This commit is contained in:
2025-09-18 08:47:11 +08:00
parent 2976f7686c
commit a015a4bc7e
3 changed files with 12 additions and 11 deletions

View File

@@ -8,6 +8,9 @@ from utils.websocket_server import room_manager
class DeepSortModel:
"""
目标追踪项模型先通过yolo对目标进行检测然后通过deepsort进行目标追踪
"""
def __init__(self, pt_url):
self.model = YOLO(pt_url)
@@ -49,8 +52,8 @@ class DeepSortModel:
# 绘制跟踪结果
for track in tracks:
if not track.is_confirmed():
active_tracks.append(track)
continue
active_tracks.append(track)
track_id = track.track_id
track_cls = str(track.det_class)
ltrb = track.to_ltrb()