From a38b27a78dae3375bac89b53bd65e88bd7877b59 Mon Sep 17 00:00:00 2001
From: zqc <835569504@qq.com>
Date: Fri, 17 Apr 2026 10:17:31 +0800
Subject: [PATCH] =?UTF-8?q?indoor=5Fbiz=E5=A2=9E=E5=8A=A0=E5=B8=A7?=
 =?UTF-8?q?=E5=9B=9E=E6=BA=AF=EF=BC=88=E6=9C=AA=E6=B5=8B=E8=AF=95=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 biz/prison/indoor_biz.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/biz/prison/indoor_biz.py b/biz/prison/indoor_biz.py
index 62e56d9..ae9f38e 100644
--- a/biz/prison/indoor_biz.py
+++ b/biz/prison/indoor_biz.py
@@ -3,6 +3,8 @@ import numpy as np
 import time
 # import requests
 from collections import deque
+
+from biz.base_detector import BaseDetector
 from biz.base_frame_processor import BaseFrameProcessorWorker
 from algorithm.common.npu_yolo_onnx_person_car_phone import YOLOv8_ONNX
 from yolox.tracker.byte_tracker import BYTETracker
@@ -30,8 +32,9 @@ DEFAULT_DOOR_ROIS = {
 }
 
 # ==================================================================================
-class PrisonerDoorDetector:
+class PrisonerDoorDetector(BaseDetector):
     def __init__(self, params=None):
+        super().__init__()
         self.params = params or {}
 
         # 0. 从params解析ROI配置，无则使用默认值
@@ -80,6 +83,11 @@ class PrisonerDoorDetector:
         # 距离阈值（用于匹配检测框和已有目标）
         self.distance_threshold = 100  # 像素距离
 
+        buffer_seconds = 3  # 最大回溯3秒
+        self.init_frame_buffer(buffer_seconds, RTSP_FPS)
+        self.detect_rollback_time = 0.9 # 警报帧回溯时间（秒）
+
+
     def compute_center_distance(self, box1, box2):
         """计算两个框中心点的欧氏距离"""
         cx1 = (box1[0] + box1[2]) / 2
@@ -391,11 +399,13 @@ class PrisonerDoorDetector:
                         #     timestamp=timestamp,
                         #     entry_frame=entry_frame
                         # )
+                        alert_frame = self.find_target_frame(timestamp - self.detect_rollback_time)
 
                         current_frame_alerts.append({
                             "time": timestamp,
                             "camera_id": camera_id,
                             "action": "Indoor Violation",
+                            'image': alert_frame,
                             "prisoner_track_id": target_id,
                             "disappear_roi": target_info['current_roi'],
                             "last_cx": round(target_info['last_cxcy'][0], 2),
@@ -464,6 +474,8 @@ class PrisonerDoorDetector:
                     (20, self.frame_height - 50),
                     cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 255, 0), 2)
 
+        self.append_frame(frame, timestamp)
+
         return {"image": frame, "alerts": current_frame_alerts}