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}")
                 # 等待一段时间后重试

core/rtmp.py

@@ -2,6 +2,17 @@ import asyncio
 import logging
 import cv2
 import time
+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"
+)
 
 # 配置日志（与WHEP代码保持一致的日志风格）
 logging.basicConfig(level=logging.INFO)
@@ -67,6 +78,15 @@ async def rtmp_pull_video_stream(rtmp_url):
             print(f"  帧尺寸: {width}x{height}")
             print(f"  配置帧率: {fps:.2f} FPS")
 
+            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("图片中未检测到违禁词")
             # 7. 每100帧统计一次实际接收帧率（补充性能监控，与原RTMP示例逻辑一致）
             if frame_count % 100 == 0:
                 elapsed_time = time.time() - start_time