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