wsy182
08b6543b79
- 添加视频处理性能计时和统计功能 - 实现帧处理时间监控和慢帧警告 - 添加音频文件静音修剪功能 - 优化Windows平台音频播放实现 - 调整默认日志输出频率减少冗余信息 - 修复MediaPipe GPU委托在Windows上的兼容性问题
88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
|
|
|
|
@dataclass
|
|
class ServerConfig:
|
|
"""WebSocket服务器配置"""
|
|
host: str = "0.0.0.0"
|
|
port: int = 8765
|
|
max_ws_size: int = 10_485_760
|
|
|
|
|
|
@dataclass
|
|
class VideoConfig:
|
|
"""视频帧处理配置"""
|
|
process_every_n_frames: int = 2
|
|
log_every_n_frames: int = 30
|
|
perf_log_every_n_frames: int = 30
|
|
slow_frame_ms: float = 100.0
|
|
|
|
|
|
@dataclass
|
|
class ModelConfig:
|
|
"""姿态检测模型配置"""
|
|
path: str = ""
|
|
prefer_gpu: bool = True
|
|
|
|
@property
|
|
def resolved_path(self) -> str:
|
|
"""返回模型文件的绝对路径"""
|
|
if self.path:
|
|
return self.path
|
|
return str(Path(__file__).resolve().parent.parent / "pose_models" / "pose_landmarker_full.task")
|
|
|
|
|
|
@dataclass
|
|
class DeadBugConfig:
|
|
"""死虫式（Dead Bug）运动检测配置"""
|
|
visibility_threshold: float = 0.45
|
|
extension_confirm_frames: int = 4
|
|
reset_confirm_frames: int = 3
|
|
|
|
|
|
@dataclass
|
|
class AudioConfig:
|
|
"""语音播报配置"""
|
|
rep_announcer_enabled: bool = True
|
|
rep_announcer_rate: int = 185
|
|
rep_announcer_volume: float = 1.0
|
|
rep_max_count: int = 200
|
|
rep_audio_dir: str = ""
|
|
trim_leading_silence: bool = True
|
|
trim_silence_threshold: int = 500
|
|
trim_silence_padding_ms: int = 20
|
|
|
|
@property
|
|
def resolved_audio_dir(self) -> Path:
|
|
"""返回语音文件目录的绝对路径"""
|
|
if self.rep_audio_dir:
|
|
return Path(self.rep_audio_dir)
|
|
return Path(__file__).resolve().parent.parent / "resources" / "audio" / "reps"
|
|
|
|
|
|
@dataclass
|
|
class LoggingConfig:
|
|
"""日志配置"""
|
|
dir: str = "logs"
|
|
rotation: str = "20 MB"
|
|
retention: str = "14 days"
|
|
|
|
@property
|
|
def dir_path(self) -> Path:
|
|
"""返回日志目录的绝对路径"""
|
|
return Path(__file__).resolve().parent.parent / self.dir
|
|
|
|
|
|
@dataclass
|
|
class AppConfig:
|
|
"""应用总配置，聚合所有子配置"""
|
|
server: ServerConfig = field(default_factory=ServerConfig)
|
|
video: VideoConfig = field(default_factory=VideoConfig)
|
|
model: ModelConfig = field(default_factory=ModelConfig)
|
|
dead_bug: DeadBugConfig = field(default_factory=DeadBugConfig)
|
|
audio: AudioConfig = field(default_factory=AudioConfig)
|
|
logging: LoggingConfig = field(default_factory=LoggingConfig)
|