yipai-tech/SupervisorAI
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

移除不必要的数据

Browse Source
This commit is contained in:
zqc
2026-02-04 16:42:05 +08:00
parent 4625f3dfb8
commit d9d681181e
1 changed files with 46 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
hls_service_ws_kadian.py
View File

@@ -157,7 +157,6 @@ class TSReaderWorker(threading.Thread):
logger.error("[ERROR] Failed to initialize playback") logger.error("[ERROR] Failed to initialize playback")
return return
frame_index = 0
while not self.stop_event.is_set(): while not self.stop_event.is_set():
try: try:
@@ -166,7 +165,6 @@ class TSReaderWorker(threading.Thread):
time.sleep(0.2) time.sleep(0.2)
continue continue
if frame_index >= len(current_frames):
# 当前分片播放完毕，查找下一个分片 # 当前分片播放完毕，查找下一个分片
next_segment = self.find_next_segment() next_segment = self.find_next_segment()
@@ -194,21 +192,11 @@ class TSReaderWorker(threading.Thread):
time.sleep(0.5) time.sleep(0.5)
continue continue
# 获取当前帧 # 一次性将所有帧放入队列
frame_data = current_frames[frame_index] for frame_data in current_frames:
frame = frame_data['frame']
pts_ms = frame_data['pts_ms']
# 计算预期的播放时间
expected_play_time = self.start_time + (pts_ms - self.base_pts) / 1000.0
# 准备帧数据
item = { item = {
"camera_id": self.camera_cfg.id, "frame": frame_data['frame'],
"camera_name": self.camera_cfg.name, "pts_ms": frame_data['pts_ms']
"timestamp": expected_play_time, # 使用预期的播放时间
"frame": frame,
"pts_ms": pts_ms
} }
try: try:
@@ -216,15 +204,14 @@ class TSReaderWorker(threading.Thread):
if self.frame_queue.full(): if self.frame_queue.full():
try: try:
self.frame_queue.get_nowait() self.frame_queue.get_nowait()
self.frame_queue.task_done()
except queue.Empty: except queue.Empty:
pass pass
self.frame_queue.put(item, timeout=0.5) self.frame_queue.put(item, timeout=0.5)
frame_index += 1
except queue.Full: except queue.Full:
logger.warning(f"[WARN] Queue full, dropping frame from {self.camera_cfg.name}") logger.warning(f"[WARN] Queue full, dropping frame from {self.camera_cfg.name}")
break # 如果队列满，停止放入更多帧
except Exception as e: except Exception as e:
logger.error(f"[ERROR] TSReaderWorker loop error: {e}") logger.error(f"[ERROR] TSReaderWorker loop error: {e}")
@@ -299,8 +286,8 @@ class HLSFrameProcessor(threading.Thread):
# 时间同步良好，放入raw_queue # 时间同步良好，放入raw_queue
item = { item = {
"camera_id": frame_data["camera_id"], "camera_id": self.camera_cfg.id,
"camera_name": frame_data["camera_name"], "camera_name": self.camera_cfg.name,
"timestamp": current_time, "timestamp": current_time,
"frame": frame_data["frame"], "frame": frame_data["frame"],
} }
@@ -317,7 +304,7 @@ class HLSFrameProcessor(threading.Thread):
self.raw_queue.put(item, timeout=0.5) self.raw_queue.put(item, timeout=0.5)
except queue.Full: except queue.Full:
logger.warning(f"[WARN] Raw queue full, dropping frame from {frame_data['camera_name']}") logger.warning(f"[WARN] Raw queue full, dropping frame from {self.camera_cfg.name}")
# 标记帧处理完成 # 标记帧处理完成
self.frame_queue.task_done() self.frame_queue.task_done()