This commit is contained in:
2025-02-24 15:28:57 +08:00
parent f912c98e16
commit b173e70ae7
2 changed files with 1 additions and 42 deletions

2
.gitignore vendored
View File

@@ -1,7 +1,7 @@
# ---> Python
_backend/evaluate/eval_rag_dataset/*
_backend/history/*
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]

View File

@@ -266,52 +266,11 @@ async def chat(websocket: WebSocket):
# 设置缓冲区大小
BUFFER_SIZE = 10
frame_buffer = deque(maxlen=BUFFER_SIZE)
RTSP_STREAMS = {
"camera1": "rtsp://admin:@192.168.1.13:554/live",
"camera2": "rtsp://admin:@192.168.1.10:554/live",
}
# @app.websocket("/video_stream/{camera_id}")
# async def websocket_endpoint(websocket: WebSocket, camera_id: str):
# await websocket.accept()
# cap = cv2.VideoCapture(RTSP_STREAMS[camera_id])
# fps = 15
# cap.set(cv2.CAP_PROP_FPS, 15)
# frame_interval = 1 / fps if fps > 0 else 0.1 # 计算帧间隔时间
# async def capture_frames():
# while True:
# ret, frame = cap.read()
# if not ret:
# break
# _, buffer = cv2.imencode('.jpg', frame)
# jpeg_frame = base64.b64encode(buffer).decode('utf-8')
# frame_buffer.append(jpeg_frame)
# await asyncio.sleep(frame_interval)
# async def send_frames():
# while True:
# if frame_buffer:
# frame = frame_buffer.popleft()
# await websocket.send_text(frame)
# await asyncio.sleep(frame_interval)
# # 启动帧捕获和发送任务
# capture_task = asyncio.create_task(capture_frames())
# send_task = asyncio.create_task(send_frames())
# try:
# await asyncio.gather(capture_task, send_task)
# except asyncio.CancelledError:
# pass
# finally:
# cap.release()
frame_buffer = deque()
@app.websocket("/video_stream/{camera_id}")
async def websocket_endpoint(websocket: WebSocket, camera_id: str):
await websocket.accept()