ocr1.0

core/rtc.py

@@ -2,13 +2,24 @@ import asyncio
 import logging
 from aiortc import RTCPeerConnection, RTCSessionDescription
 import aiohttp
+from ocr.ocr_violation_detector import OCRViolationDetector
+
+import logging
+
+# 创建检测器实例
+detector = OCRViolationDetector(
+        forbidden_words_path=r"D:\Git\bin\video\ocr\forbidden_words.txt",
+        ocr_confidence_threshold=0.7,
+        log_level=logging.INFO,
+        log_file="ocr_detection.log"
+)
 
 # 配置日志
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("whep_video_puller")
 
 
-async def whep_pull_video_stream(whep_url):
+async def whep_pull_video_stream(ip,whep_url):
     """
     通过WHEP从指定URL拉取视频流并在收到每一帧时打印消息
 
@@ -60,6 +71,15 @@ async def whep_pull_video_stream(whep_url):
                 if hasattr(frame, 'pts'):
                     print(f"  显示时间戳: {frame.pts}")
 
+                has_violation, violations, confidences = OCRViolationDetector.detect(frame)
+
+                # 输出检测结果
+                if has_violation:
+                    detector.logger.info(f"在图片中检测到 {len(violations)} 个违禁词:")
+                    for word, conf in zip(violations, confidences):
+                        detector.logger.info(f"- {word} (置信度: {conf:.4f})")
+                else:
+                    detector.logger.info("图片中未检测到违禁词")
             except Exception as e:
                 print(f"接收帧时出错: {e}")
                 # 等待一段时间后重试