sun15684712291/aicheckv2-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

系统配置问题

Browse Source
This commit is contained in:
sunyg
2025-09-18 08:47:11 +08:00
parent 2976f7686c
commit a015a4bc7e
3 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
algo/deep_sort.py
View File

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

1
application/config/development.py
View File

@@ -58,4 +58,3 @@ images_url = r'D:\syg\images'
datasets_url = r'D:\syg\yolov5\datasets' datasets_url = r'D:\syg\yolov5\datasets'
runs_url = r'D:\syg\yolov5\runs' runs_url = r'D:\syg\yolov5\runs'
detect_url = r'D:\syg\yolov5\detect' detect_url = r'D:\syg\yolov5\detect'
yolo_url = r'D:\syg\workspace\aicheckv2\yolov5'

15
application/config/production.py
View File

@@ -11,20 +11,20 @@ Mysql 数据库配置项
连接引擎官方文档：https://www.osgeo.cn/sqlalchemy/core/engines.html 连接引擎官方文档：https://www.osgeo.cn/sqlalchemy/core/engines.html
数据库链接配置说明：mysql+asyncmy://数据库用户名:数据库密码@数据库地址:数据库端口/数据库名称 数据库链接配置说明：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://:密码@地址:端口/数据库名称" 格式："redis://:密码@地址:端口/数据库名称"
""" """
REDIS_DB_ENABLE = True 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 数据库配置
格式：mongodb://用户名:密码@地址:端口/?authSource=数据库名称 格式：mongodb://用户名:密码@地址:端口/?authSource=数据库名称
""" """
MONGO_DB_ENABLE = True MONGO_DB_ENABLE = False
MONGO_DB_NAME = "aicheck" MONGO_DB_NAME = "aicheck"
MONGO_DB_URL = f"mongodb://localhost:27017/?authSource={MONGO_DB_NAME}" MONGO_DB_URL = f"mongodb://localhost:27017/?authSource={MONGO_DB_NAME}"
@@ -53,8 +53,7 @@ runs_url 训练结果文件地址
detect_url 推理集合文件地址 detect_url 推理集合文件地址
yolo_url yolo执行文件所在的文件路径 yolo_url yolo执行文件所在的文件路径
""" """
images_url = f'/home/aicheckv2/images' images_url = f'/home/syg/aicheckv2.1/images'
datasets_url = f'/home/aicheckv2/yolov5/datasets' datasets_url = f'/home/syg/aicheckv2.1/datasets'
runs_url = f'/home/aicheckv2/yolov5/runs' runs_url = f'/home/syg/aicheckv2.1/runs'
detect_url = f'/home/aicheckv2/yolov5/detect' detect_url = f'/home/syg/aicheckv2.1/detect'
yolo_url = f'/home/aicheckv2/backend/yolov5'