yipai-tech/SupervisorAI
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改算法文件路径改为common

Browse Source
This commit is contained in:
zqc
2026-02-26 11:39:14 +08:00
parent cd13709998
commit 7ce50cfd4f
11 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
algorithm/checkpoint/__init__.py → algorithm/common/__init__.py
View File

0
algorithm/checkpoint/npu_yolo_onnx_person_car_phone.py → algorithm/common/npu_yolo_onnx_person_car_phone.py
View File

0
algorithm/checkpoint/npu_yolo_pose_onnx.py → algorithm/common/npu_yolo_pose_onnx.py
View File

4
biz/checkpoint/checkpoint_biz.py
View File

@@ -8,8 +8,8 @@ import time
import queue import queue
# -------------------------- Kadian 检测相关导入 -------------------------- # -------------------------- Kadian 检测相关导入 --------------------------
from algorithm.checkpoint.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机） from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机）
from algorithm.checkpoint.npu_yolo_pose_onnx import YOLOv8_Pose_ONNX # Pose 专用模型 from algorithm.common.npu_yolo_pose_onnx import YOLOv8_Pose_ONNX # Pose 专用模型
from yolox.tracker.byte_tracker import BYTETracker from yolox.tracker.byte_tracker import BYTETracker
from utils.logger import get_logger from utils.logger import get_logger

2
biz/prison/prison_biz.py
View File

@@ -8,7 +8,7 @@ import queue
import requests import requests
# -------------------------- Kadian 检测相关导入 -------------------------- # -------------------------- Kadian 检测相关导入 --------------------------
from algorithm.checkpoint.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机） from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机）
from yolox.tracker.byte_tracker import BYTETracker from yolox.tracker.byte_tracker import BYTETracker

3
hls_service_ws_kadian.py
View File

@@ -12,7 +12,8 @@ import glob
from dataclasses import dataclass from dataclasses import dataclass
from biz.checkpoint.checkpoint_biz import KadianDetector, RTSP_TARGET_FPS, ALERT_PUSH_INTERVAL, FrameProcessorWorker # from biz.checkpoint.checkpoint_biz import FrameProcessorWorker
from biz.prison.prison_biz import FrameProcessorWorker
from utils.web_socket_sender import WebSocketSender from utils.web_socket_sender import WebSocketSender
from utils.logger import get_logger from utils.logger import get_logger

2
rtsp_service_ws_0108.py
View File

@@ -31,7 +31,7 @@ except Exception as e:
# -------------------------- Kadian 检测相关导入 -------------------------- # -------------------------- Kadian 检测相关导入 --------------------------
from algorithm.checkpoint.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机） from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机）
from yolox.tracker.byte_tracker import BYTETracker from yolox.tracker.byte_tracker import BYTETracker

2
rtsp_service_ws_Zhihuishi.py
View File

@@ -18,7 +18,7 @@ from typing import Dict, Any
from test_cam import get_camera_preview_url from test_cam import get_camera_preview_url
# -------------------------- Kadian 检测相关导入 -------------------------- # -------------------------- Kadian 检测相关导入 --------------------------
from algorithm.checkpoint.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机） from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机）
from yolox.tracker.byte_tracker import BYTETracker from yolox.tracker.byte_tracker import BYTETracker

2
rtsp_service_ws_Zoulang.py
View File

@@ -19,7 +19,7 @@ from typing import Dict, Any
from test_cam import get_camera_preview_url from test_cam import get_camera_preview_url
# -------------------------- Kadian 检测相关导入 -------------------------- # -------------------------- Kadian 检测相关导入 --------------------------
from algorithm.checkpoint.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机） from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机）
from yolox.tracker.byte_tracker import BYTETracker from yolox.tracker.byte_tracker import BYTETracker

2
rtsp_service_ws_kadian.py
View File

@@ -12,7 +12,7 @@ import yaml
from dataclasses import dataclass from dataclasses import dataclass
from biz.checkpoint.checkpoint_biz import KadianDetector, RTSP_TARGET_FPS, ALERT_PUSH_INTERVAL, FrameProcessorWorker from biz.checkpoint.checkpoint_biz import FrameProcessorWorker
from test_cam import get_camera_preview_url from test_cam import get_camera_preview_url
from utils.web_socket_sender import WebSocketSender from utils.web_socket_sender import WebSocketSender

2
rtsp_service_ws_kanshousuo.py
View File

@@ -16,7 +16,7 @@ from dataclasses import dataclass
from typing import Dict, Any from typing import Dict, Any
# -------------------------- Kadian 检测相关导入 -------------------------- # -------------------------- Kadian 检测相关导入 --------------------------
from algorithm.checkpoint.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机） from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX # 主检测模型（人/车/后备箱/手机）
from yolox.tracker.byte_tracker import BYTETracker from yolox.tracker.byte_tracker import BYTETracker