视频监控人员离岗睡岗检测系统：从行为感知到合规管理实践

import time
from typing import Dict, Optional, Tuple

class DutyPostureMonitor:
    def __init__(self, away_threshold_sec=300, sleep_threshold_sec=60):
        self.away_threshold = away_threshold_sec      # 离岗超时阈值（秒）
        self.sleep_threshold = sleep_threshold_sec    # 睡岗确认阈值（秒）
        self.state = {
            "last_seen_time": None,
            "sleep_start_time": None,
            "is_away": False,
            "is_sleeping": False
        }

    def update(self, person_in_roi: bool, is_sleep_posture: bool) -> Optional[str]:
        """
        :param person_in_roi: 当前帧人员是否在岗位ROI内
        :param is_sleep_posture: 当前帧是否为睡岗姿态
        :return: 告警类型字符串，无告警返回None
        """
        now = time.time()
        alert_type = None

        # 更新最后出现时间
        if person_in_roi:
            self.state["last_seen_time"] = now
            self.state["is_away"] = False
        else:
            if self.state["last_seen_time"] is not None:
                away_duration = now - self.state["last_seen_time"]
                if away_duration > self.away_threshold and not self.state["is_away"]:
                    self.state["is_away"] = True
                    alert_type = "AWAY_OVERTIME"

        # 睡岗状态机
        if person_in_roi and is_sleep_posture:
            if self.state["sleep_start_time"] is None:
                self.state["sleep_start_time"] = now
            elif now - self.state["sleep_start_time"] > self.sleep_threshold:
                if not self.state["is_sleeping"]:
                    self.state["is_sleeping"] = True
                    alert_type = "SLEEP_DETECTED"
        else:
            # 姿态恢复正常，重置睡岗计时
            self.state["sleep_start_time"] = None
            self.state["is_sleeping"] = False

        return alert_type

# 使用示例
# monitor = DutyPostureMonitor(away_threshold_sec=300, sleep_threshold_sec=60)
# # 模拟帧处理循环
# # alert = monitor.update(person_in_roi=True, is_sleep_posture=True)
# # if alert == "SLEEP_DETECTED": trigger_alarm("sleep")