You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
# ptz_tracker_modular/detection.py
|
|
# -*- coding: utf-8 -*-
|
|
from __future__ import annotations
|
|
import os as _os
|
|
|
|
from .config import (
|
|
INFER_IMGSZ, MODEL_CONF, TARGET_CLASSES,
|
|
W_EMA_ALPHA, DEFAULT_CLIENT_FPS, DRAW_ALL_ON_PANO,
|
|
DETECT_CONF, ALARM_CONF, RECOVER_ZOOM_OUT,
|
|
CENTER_GATE_IN, AIM_SMOOTH, AIM_P, AIM_I, AIM_D,
|
|
AUTO_STRATEGY, SHORT_LOSS_SEC, LONG_LOSS_SEC,
|
|
ABS_MIN_FALLBACK, ABS_MAX_FALLBACK,
|
|
)
|
|
# Сохранение старого имени для веса модели
|
|
YOLO_WEIGHTS = _os.getenv("YOLO_WEIGHTS", "C:/Users/Legion/Desktop/weight/electro-pribory-new.pt")
|
|
|
|
from .utils import clipf
|
|
from .state import ptz_states, detect_stats_global, detect_stats_cam, video_clients, batcher as _batcher
|
|
from .preview import publish_preview, maybe_downscale
|
|
from .postprocess import notify_detected, postprocess_control
|
|
from .events import queue_alarm
|
|
from .ptz_io import move_ptz_bounded, stop_ptz, call_autofocus
|
|
from .patrol import return_to_preset
|
|
|
|
from .streaming import MicroBatcher
|
|
from .model import yolo_forward
|
|
|
|
# Старый код иногда ждал detection.batcher — даём псевдоним на state.batcher
|
|
batcher = _batcher
|
|
|
|
__all__ = [
|
|
'INFER_IMGSZ', 'MODEL_CONF', 'TARGET_CLASSES',
|
|
'W_EMA_ALPHA', 'DEFAULT_CLIENT_FPS', 'DRAW_ALL_ON_PANO',
|
|
'DETECT_CONF','ALARM_CONF','RECOVER_ZOOM_OUT',
|
|
'CENTER_GATE_IN','AIM_SMOOTH','AIM_P','AIM_I','AIM_D',
|
|
'AUTO_STRATEGY','SHORT_LOSS_SEC','LONG_LOSS_SEC','ABS_MIN_FALLBACK','ABS_MAX_FALLBACK',
|
|
'YOLO_WEIGHTS',
|
|
'clipf', 'ptz_states','detect_stats_global','detect_stats_cam','video_clients',
|
|
'publish_preview','maybe_downscale','notify_detected','postprocess_control','queue_alarm',
|
|
'move_ptz_bounded','stop_ptz','call_autofocus','return_to_preset',
|
|
'MicroBatcher','yolo_forward','batcher'
|
|
]
|