실시간 비디오 분석을 위한 RTSP 스트림 처리 파이프라인 구축

class StreamReader:
    def __init__(self, rtsp_url: str, reconnect_delay: int = 5, watchdog_timeout: int = 10):
        self.rtsp_url = rtsp_url
        self.reconnect_delay = reconnect_delay
        self.watchdog_timeout = watchdog_timeout
        self.frame = None
        self.last_frame_time = None
        self.running = False
        self.cap = None
        self.fps = 30.0
        self._lock = threading.Lock()
        self._thread = None

    def start(self):
        self.running = True
        self._thread = threading.Thread(target=self._read_loop, daemon=True)
        self._thread.start()
        return self

class FireSmokeDetector:
    def __init__(self, api_url: str, api_key: str, model_id: str, confidence: float = 0.4):
        self.model_id = model_id
        self.confidence = confidence
        self.client = InferenceHTTPClient(
            api_url=api_url,
            api_key=api_key,
        )

    def predict(self, frame: np.ndarray) -> dict:
        try:
            result = self.client.infer(frame, model_id=self.model_id)
            result["predictions"] = [
                p for p in result.get("predictions", []) if p["confidence"] >= self.confidence
            ]
            return result