可以成功动态更换yolo模型并重启服务生效
This commit is contained in:
@ -1,4 +1,3 @@
|
||||
import os
|
||||
import datetime
|
||||
from pathlib import Path
|
||||
|
||||
@ -9,19 +8,19 @@ def create_directory_structure():
|
||||
# 1. 创建根目录下的resource文件夹(存在则跳过,不覆盖子内容)
|
||||
resource_dir = Path("resource")
|
||||
resource_dir.mkdir(exist_ok=True)
|
||||
print(f"确保resource目录存在: {resource_dir.absolute()}")
|
||||
# print(f"确保resource目录存在: {resource_dir.absolute()}")
|
||||
|
||||
# 2. 在resource下创建dect文件夹
|
||||
dect_dir = resource_dir / "dect"
|
||||
dect_dir.mkdir(exist_ok=True)
|
||||
print(f"确保dect目录存在: {dect_dir.absolute()}")
|
||||
# print(f"确保dect目录存在: {dect_dir.absolute()}")
|
||||
|
||||
# 3. 在dect下创建三个模型文件夹
|
||||
model_dirs = ["ocr", "face", "yolo"]
|
||||
for model in model_dirs:
|
||||
model_dir = dect_dir / model
|
||||
model_dir.mkdir(exist_ok=True)
|
||||
print(f"确保{model}模型目录存在: {model_dir.absolute()}")
|
||||
# print(f"确保{model}模型目录存在: {model_dir.absolute()}")
|
||||
|
||||
# 4. 调用外部方法获取所有客户端IP地址
|
||||
try:
|
||||
@ -59,7 +58,7 @@ def create_directory_structure():
|
||||
|
||||
# 递归创建目录(存在则跳过,不覆盖)
|
||||
month_dir.mkdir(parents=True, exist_ok=True)
|
||||
print(f"为客户端IP {ip} 创建/确保目录存在: {month_dir.absolute()}")
|
||||
# print(f"为客户端IP {ip} 创建/确保目录存在: {month_dir.absolute()}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"处理客户端IP和日期目录时发生错误: {str(e)}")
|
||||
|
@ -1,12 +1,9 @@
|
||||
import os
|
||||
import numpy as np
|
||||
import cv2
|
||||
import gc
|
||||
import time
|
||||
import threading
|
||||
from PIL import Image
|
||||
from insightface.app import FaceAnalysis
|
||||
# 假设service.face_service中get_all_face_name_with_eigenvalue可获取人脸数据
|
||||
from service.face_service import get_all_face_name_with_eigenvalue
|
||||
|
||||
# GPU状态检查支持
|
||||
|
@ -1,5 +1,3 @@
|
||||
import os
|
||||
import numpy as np
|
||||
from ultralytics import YOLO
|
||||
from service.model_service import get_current_yolo_model # 带版本校验的模型获取
|
||||
|
||||
|
Reference in New Issue
Block a user