From 1f00f8f3f76c600e92f521d4a19923904aa22578 Mon Sep 17 00:00:00 2001 From: "DESKTOP-0HKKMNF\\R7000" <19966421056@163.com> Date: Tue, 14 Apr 2026 11:20:02 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=A0=E9=99=A4=E6=8A=A5=E8=AD=A6=E7=9B=B8?= =?UTF-8?q?=E5=85=B3=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- biz/prison/indoor_biz.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/biz/prison/indoor_biz.py b/biz/prison/indoor_biz.py index 44df1ae..62e56d9 100644 --- a/biz/prison/indoor_biz.py +++ b/biz/prison/indoor_biz.py @@ -1,7 +1,7 @@ import cv2 import numpy as np import time -import requests +# import requests from collections import deque from biz.base_frame_processor import BaseFrameProcessorWorker from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX @@ -13,7 +13,7 @@ DETECT_MODEL_PATH = 'YOLO_Weight/kanshousuo.onnx' # 犯人检测onnx模型路 INPUT_SIZE = 640 # 模型输入尺寸 RTSP_FPS = 10 # 视频流目标FPS ALERT_PUSH_INTERVAL = 5 # 相同报警5秒内仅推送1次 -ALERT_PUSH_URL = "http://123.57.151.210:10000/picenter/websocket/test/process" +# ALERT_PUSH_URL = "http://123.57.151.210:10000/picenter/websocket/test/process" # 消失判定：中心点在ROI内消失后，持续无检测的帧数（1.0秒，可微调） ROI_LOST_FRAMES_THRESH = int(0.5 * RTSP_FPS) # todo: 从frame改为时间 @@ -64,7 +64,7 @@ class PrisonerDoorDetector: self.tracker = BYTETracker(TrackerArgs(), frame_rate=RTSP_FPS) # 3. 状态变量初始化 - self.last_alert_time = 0.0 # 最后报警时间（防重复推送） + # self.last_alert_time = 0.0 # 最后报警时间（防重复推送） # 犯人跟踪信息：{track_id: {'is_cx_in_roi': 中心点是否在ROI, 'lost_frames': 消失帧数, 'lost_roi': 消失的ROI名称, 'last_cxcy': 最后中心点坐标}} self.prisoner_track_info = {} self.frame_width = 0 # 帧宽度（动态获取） @@ -382,15 +382,15 @@ class PrisonerDoorDetector: if target_info['lost_frames'] >= ROI_LOST_FRAMES_THRESH: # 触发报警 - entry_frame = self.entry_frame_cache.get(target_id, frame_copy) - self.push_alert( - camera_id=camera_id, - target_id=target_id, - lost_roi=target_info['current_roi'], - last_cxcy=target_info['last_cxcy'], - timestamp=timestamp, - entry_frame=entry_frame - ) + # entry_frame = self.entry_frame_cache.get(target_id, frame_copy) + # self.push_alert( + # camera_id=camera_id, + # target_id=target_id, + # lost_roi=target_info['current_roi'], + # last_cxcy=target_info['last_cxcy'], + # timestamp=timestamp, + # entry_frame=entry_frame + # ) current_frame_alerts.append({ "time": timestamp,