From a015a4bc7e76831b7c5166aa4388a6c843aee43e Mon Sep 17 00:00:00 2001 From: sunyugang Date: Thu, 18 Sep 2025 08:47:11 +0800 Subject: [PATCH] =?UTF-8?q?=E7=B3=BB=E7=BB=9F=E9=85=8D=E7=BD=AE=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- algo/deep_sort.py | 5 ++++- application/config/development.py | 3 +-- application/config/production.py | 15 +++++++-------- 3 files changed, 12 insertions(+), 11 deletions(-) diff --git a/algo/deep_sort.py b/algo/deep_sort.py index e10b3e3..65da095 100644 --- a/algo/deep_sort.py +++ b/algo/deep_sort.py @@ -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() diff --git a/application/config/development.py b/application/config/development.py index 1eba8ae..d317cf5 100644 --- a/application/config/development.py +++ b/application/config/development.py @@ -57,5 +57,4 @@ yolo_url yolov5执行文件所在的文件路径 images_url = r'D:\syg\images' datasets_url = r'D:\syg\yolov5\datasets' runs_url = r'D:\syg\yolov5\runs' -detect_url = r'D:\syg\yolov5\detect' -yolo_url = r'D:\syg\workspace\aicheckv2\yolov5' \ No newline at end of file +detect_url = r'D:\syg\yolov5\detect' \ No newline at end of file diff --git a/application/config/production.py b/application/config/production.py index d7a5784..bbd86dc 100644 --- a/application/config/production.py +++ b/application/config/production.py @@ -11,20 +11,20 @@ Mysql 数据库配置项 连接引擎官方文档:https://www.osgeo.cn/sqlalchemy/core/engines.html 数据库链接配置说明:mysql+asyncmy://数据库用户名:数据库密码@数据库地址:数据库端口/数据库名称 """ -SQLALCHEMY_DATABASE_URL = "mysql+asyncmy://root:123456@177.8.0.7:3306/aicheck-pro" +SQLALCHEMY_DATABASE_URL = "mysql+asyncmy://root:Aicheck2025@localhost:3306/aicheck-pro" """ Redis 数据库配置 格式:"redis://:密码@地址:端口/数据库名称" """ REDIS_DB_ENABLE = True -REDIS_DB_URL = "redis://:123456@177.8.0.5:6379/1" +REDIS_DB_URL = "redis://:sdust2020@localhost:6379/1" """ MongoDB 数据库配置 格式:mongodb://用户名:密码@地址:端口/?authSource=数据库名称 """ -MONGO_DB_ENABLE = True +MONGO_DB_ENABLE = False MONGO_DB_NAME = "aicheck" MONGO_DB_URL = f"mongodb://localhost:27017/?authSource={MONGO_DB_NAME}" @@ -53,8 +53,7 @@ runs_url 训练结果文件地址 detect_url 推理集合文件地址 yolo_url yolo执行文件所在的文件路径 """ -images_url = f'/home/aicheckv2/images' -datasets_url = f'/home/aicheckv2/yolov5/datasets' -runs_url = f'/home/aicheckv2/yolov5/runs' -detect_url = f'/home/aicheckv2/yolov5/detect' -yolo_url = f'/home/aicheckv2/backend/yolov5' \ No newline at end of file +images_url = f'/home/syg/aicheckv2.1/images' +datasets_url = f'/home/syg/aicheckv2.1/datasets' +runs_url = f'/home/syg/aicheckv2.1/runs' +detect_url = f'/home/syg/aicheckv2.1/detect' \ No newline at end of file