系统配置问题
This commit is contained in:
@@ -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()
|
||||
|
Reference in New Issue
Block a user