You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.5 KiB
48 lines
1.5 KiB
# -*- coding:utf-8 -*-
|
|
# @Time : 2021/12/31 16:17
|
|
# @Author : lxc
|
|
# @File : tortsp.py
|
|
# https://github.com/aler9/rtsp-simple-server
|
|
|
|
import cv2
|
|
|
|
|
|
import subprocess as sp
|
|
import signal
|
|
import numpy as npbool
|
|
class RtstPushServer:
|
|
def __init__(self, fps=25,width=1080,height=960,rtsp='rtsp://192.168.1.182:8554/video'):
|
|
print('fps=',fps,' weitht=',width,' height=',height,' rtsp=',rtsp)
|
|
#此处换为你自己的地址
|
|
self.lrtsp_url =rtsp
|
|
|
|
# Get video information
|
|
self.lfps = fps
|
|
self.lwidth = width
|
|
self.lheight = height
|
|
self.command = ['ffmpeg',
|
|
'-y',
|
|
'-f', 'rawvideo',
|
|
'-vcodec', 'rawvideo',
|
|
'-pix_fmt', 'bgr24',
|
|
'-s', "{}x{}".format(self.lwidth, self.lheight),
|
|
'-r', str(self.lfps),
|
|
'-i', '-',
|
|
'-c:v', 'libx264',
|
|
# '-c:v', 'libx265',
|
|
'-pix_fmt', 'yuv420p',
|
|
'-preset', 'ultrafast',
|
|
'-f', 'rtsp',
|
|
self.lrtsp_url]
|
|
self.p = sp.Popen(self.command, stdin=sp.PIPE)
|
|
def write(self,frame):
|
|
|
|
|
|
if frame is None:
|
|
print("frame is None")
|
|
return
|
|
# frame = 你的图像处理的函数(frame)
|
|
self.p.stdin.write(frame.tostring())
|
|
# cv2.imshow('img',frame)
|
|
# if cv2.waitKey(1) == ord('q'): # q to quit
|
|
# raise StopIteration
|
|
|