基于FastAPI实现视频流传输包含推流拉流(极简版)

ryan润

已于 2023-09-02 21:44:42 修改

阅读量1.7k

点赞数 1

分类专栏： python 文章标签： python fastapi 视频 opencv

于 2023-04-26 22:08:15 首次发布

本文链接：https://blog.csdn.net/m0_62579137/article/details/130395619

版权

python 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

FastAPI简易视频流传输

运行效果如下:
在这里插入图片描述

服务端

在服务器/本地运行以下代码:

运行后访问服务IP地址:8000可直接查看视频流.

import uvicorn
from fastapi import FastAPI, Request, WebSocket
from fastapi.responses import HTMLResponse, StreamingResponse
import cv2
import numpy as np

app = FastAPI()

# 用于存储客户端WebSocket连接的列表
client_websockets = []

# 客户端WebSocket连接的路由
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    client_websockets.append(websocket)
    try:
        while True:
            # 接收并忽略来自客户端的所有消息
            await websocket.receive()
    finally:
        # 从列表中删除已断开的WebSocket连接
        client_websockets.remove(websocket)

# 用于接收客户端上传的视频流并推送到所有WebSocket连接
@app.post("/upload")
async def upload_video(request: Request):
    # 从请求正文中读取视频流
    content = await request.body()
    nparr = np.frombuffer(content, np.uint8)
    frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)

    # 将视频帧转换为JPEG格式
    ret, buffer = cv2.imencode('.jpg', frame)
    jpg_as_text = buffer.tobytes()

    # 推送视频帧到所有WebSocket连接
    for websocket in client_websockets:
        await websocket.send_bytes(jpg_as_text)

    # 返回成功响应
    return {"status": "ok"}

# 用于提供前端查看视频的HTML页面
@app.get("/")
async def video_viewer():
    return HTMLResponse("""
        <html>
            <head>
                <title>Video Viewer</title>
            </head>
            <body>
                <h1>Real-time Video Viewer</h1>
                <img id="video_frame" src="#" alt="Video Feed" width="640" height="480"/>
                <script>
                    var video_frame = document.getElementById("video_frame");
                    var ws = new WebSocket("ws://" + window.location.host + "/ws");
                    ws.binaryType = "arraybuffer";
                    ws.onmessage = function(event) {
                        var arrayBuffer = event.data;
                        var blob = new Blob([arrayBuffer], {type: "image/jpeg"});
                        var url = URL.createObjectURL(blob);
                        video_frame.src = url;
                    };
                </script>
            </body>
        </html>
    """)

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)

客户端

要在服务器开启之后运行, 记得改ip.

import requests
import cv2
import time
from requests import ReadTimeout, ConnectTimeout
from threading import Thread
# 使用场景多数是要根据其它信息控制是否开启摄像头, 又因为要在大循环里运行, 因此继承线程类方便管理
class VideoThread(Thread):
    def run(self) -> None:
        self.cap = cv2.VideoCapture(0)  # 从摄像头读取视频
        self.should_stop = False  # 标志位用于设置线程退出
        while not self.should_stop:
            ret, frame = self.cap.read()
            if not ret:
                break
            # 在客户端显示视频流
            cv2.imshow('Video Feed', frame)
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
            _, img_encoded = cv2.imencode(".jpg", frame)  # 将帧转换为JPEG格式
            try:
                # 发送到服务器, 设置超时时间防止阻塞, 如果服务器性能一般般尽量不要发太快
                response = requests.post("http://你的服务端的ip:8000/upload", data=img_encoded.tostring(),
                                         timeout=0.05)  # 发送图像数据
                if response.status_code != 200:
                    print("接口名字是不是写错了")
            except ReadTimeout:
                # 因为设置了超时很容易网络原因超时, 丢几帧画面是可接受的
                pass
            except ConnectTimeout:
                print("服务端是不是还没开起来?")
                time.sleep(2)
        self.cap.release()
        cv2.destroyAllWindows()
    def quit(self):
        try:
            self.cap.release()
            cv2.destroyAllWindows()
        except Exception:
            # 万一出了什么异常情况也清理掉占用的资源
            pass
    def stop(self):
        self.should_stop = True

a=VideoThread()  #开启传视频
a.start()
time.sleep(60)
a.stop()

a=VideoThread()
a.start()
time.sleep(30)
a.stop()