rtsp拉流+yolov8推理+rtsp推流

最新推荐文章于 2025-02-07 17:04:45 发布

Y-Nmin

最新推荐文章于 2025-02-07 17:04:45 发布

阅读量1.7k

点赞数 15

文章标签： 1024程序员节深度学习计算机视觉

本文链接：https://blog.csdn.net/weixin_48588373/article/details/143207686

版权

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

文章目录

前言`
一、实现步骤
二、代码
总结

前言`

rtsp服务+yolov8实现实时视频流推理及前端调用rtsp地址展示推流后视频

提示：以下是本篇文章正文内容，下面案例可供参考

一、实现步骤

前提：已安装好ffmpeg和rtsp
1.rtsp拉取视频流（本地或局域网下其他服务器中的视频流）；
2.yolov8推理图片帧；
3.rtsp推流，将推理后的图片帧推流至rtsp服务器；
4.VLC拉流测试，测试推理后的图片是否推流成功；

二、代码

1.包导入

# coding=gbk
import cv2
import numpy as np
import subprocess as sp
from ultralytics1.inference_interface_function.air_ground_det import air_ground_det_infer
import requests
import time

2.rtsp拉流

src = 'rtsp://127.0.0.1:8554/stream'
dst = 'rtsp://127.0.0.1:8554/res'
cap = cv2.VideoCapture(src)
#获取原视频高宽和帧率
fps = int(cap.get(cv2.CAP_PROP_FPS))
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

3.定义ffmpeg推流命令

command = [
        'ffmpeg',
        '-y',
        '-f', 'rawvideo',
        '-vcodec', 'rawvideo',
        '-pix_fmt', 'bgr24',
        '-s', "{}x{}".format(width, height),
        '-r', str(fps),
        '-i', '-'