wsy182
6dee2a2ff3
- 调整视频处理频率从每帧处理改为每2帧处理 - 添加膝角趋势平滑算法减少单帧抖动误判 - 改进对角伸展检测逻辑支持准备位手臂上举 - 优化状态机确保严格回到准备姿态才计数 - 添加姿态丢失时的候选帧清理机制 - 更新音频文件生成路径至resources目录 - 改进macOS音频生成使用AIFF格式提高质量 - 添加详细的帧处理日志输出间隔配置
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from app.exercises.dead_bug.types import DeadBugMetrics, Point
|
|
|
|
|
|
def has_required_visibility(landmarks: list[Point], required_indices: tuple[int, ...], visibility_threshold: float) -> bool:
|
|
"""检查所有必需关键点的可见度是否高于阈值"""
|
|
return all(landmarks[i].visibility >= visibility_threshold for i in required_indices)
|
|
|
|
|
|
def detect_diagonal_extension(metrics: DeadBugMetrics) -> str | None:
|
|
"""检测对角伸展（腿部只允许单侧伸展，手臂允许准备位上举带来的识别重叠）"""
|
|
if metrics.left_leg_extended and metrics.right_leg_extended:
|
|
return None
|
|
|
|
if metrics.right_leg_extended and metrics.left_arm_extended:
|
|
return "left_arm_right_leg"
|
|
if metrics.left_leg_extended and metrics.right_arm_extended:
|
|
return "right_arm_left_leg"
|
|
return None
|
|
|
|
|
|
def is_ready_position(metrics: DeadBugMetrics) -> bool:
|
|
"""判断是否处于准备姿态（双膝弯曲且双腿未伸展；dead bug 准备位允许手臂上举）"""
|
|
knees_bent = metrics.left_knee_angle <= 140 and metrics.right_knee_angle <= 140
|
|
legs_not_extended = not metrics.left_leg_extended and not metrics.right_leg_extended
|
|
return knees_bent and legs_not_extended
|