qiyun996
3 years ago
5 changed files with 109 additions and 113 deletions
@ -0,0 +1,48 @@ |
|||
# -*- 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 |
|||
@ -1,109 +1,47 @@ |
|||
# -*- coding:utf-8 -*- |
|||
# @Time : 2021/11/4 10:17 |
|||
# @Author : JulyLi |
|||
# @File : rtsp2.py |
|||
# @Software: PyCharm |
|||
# @Time : 2021/12/3010:17 |
|||
# @Author : lxc |
|||
# @File : tortsp.py |
|||
# https://github.com/aler9/rtsp-simple-server |
|||
|
|||
import cv2 |
|||
import gi |
|||
import sys |
|||
import json |
|||
import time |
|||
import signal |
|||
import numpy as np |
|||
|
|||
gi.require_version('Gst', '1.0') |
|||
gi.require_version('GstRtspServer', '1.0') |
|||
from gi.repository import Gst, GstRtspServer, GObject |
|||
|
|||
#cv2.namedWindow('video_realtime_face', cv2.WINDOW_NORMAL) |
|||
|
|||
def to_node(type, message): |
|||
# convert to json and print (node helper will read from stdout) |
|||
try: |
|||
print(json.dumps({type: message})) |
|||
except Exception: |
|||
pass |
|||
# stdout has to be flushed manually to prevent delays in the node helper communication |
|||
sys.stdout.flush() |
|||
|
|||
to_node("status", "Facerecognition started...") |
|||
|
|||
def shutdown(self, signum): |
|||
to_node("status", 'Shutdown: Cleaning up camera...') |
|||
quit() |
|||
|
|||
signal.signal(signal.SIGINT, shutdown) |
|||
|
|||
|
|||
class SensorFactory(GstRtspServer.RTSPMediaFactory): |
|||
def __init__(self, **properties): |
|||
super(SensorFactory, self).__init__(**properties) |
|||
self.cap = cv2.VideoCapture("rtsp://admin:admin123@192.168.2.190:554/sub") |
|||
# self.cap = cv2.VideoCapture("shmsrc socket-path=/tmp/foo2 ! video/x-raw, format=BGR ,width=1920,height=1080,framerate=30/1 ! videoconvert ! video/x-raw, format=BGR ! appsink") |
|||
#self.cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920) |
|||
#self.cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080) |
|||
self.number_frames = 0 |
|||
self.fps = 30.0 |
|||
self.duration = 1 / self.fps * Gst.SECOND # duration of a frame in nanoseconds |
|||
self.launch_string = 'appsrc name=source is-live=true block=true format=GST_FORMAT_TIME ' \ |
|||
'caps=video/x-raw,format=BGR,width=1920,height=1080,framerate=30/1 ' \ |
|||
'! videoconvert ! video/x-raw,format=I420 ' \ |
|||
'! x264enc speed-preset=ultrafast tune=zerolatency threads=4 ' \ |
|||
'! rtph264pay config-interval=1 name=pay0 pt=96' |
|||
|
|||
def on_need_data(self, src, lenght): |
|||
if self.cap.isOpened(): |
|||
ret, frame = self.cap.read() |
|||
if ret: |
|||
#cv2.imshow("video_realtime_face", frame) |
|||
#if cv2.waitKey(1) & 0xFF == ord('q'): |
|||
# return |
|||
data = frame.tostring() |
|||
buf = Gst.Buffer.new_allocate(None, len(data), None) |
|||
buf.fill(0, data) |
|||
buf.duration = self.duration |
|||
timestamp = self.number_frames * self.duration |
|||
buf.pts = buf.dts = int(timestamp) |
|||
buf.offset = timestamp |
|||
self.number_frames += 1 |
|||
retval = src.emit('push-buffer', buf) |
|||
print('pushed buffer, frame {}, duration {} ns, durations {} s'.format(self.number_frames, |
|||
self.duration, |
|||
self.duration / Gst.SECOND)) |
|||
if retval != Gst.FlowReturn.OK: |
|||
print(retval) |
|||
|
|||
def do_create_element(self, url): |
|||
return Gst.parse_launch(self.launch_string) |
|||
|
|||
def do_configure(self, rtsp_media): |
|||
self.number_frames = 0 |
|||
appsrc = rtsp_media.get_element().get_child_by_name('source') |
|||
appsrc.connect('need-data', self.on_need_data) |
|||
|
|||
|
|||
class GstServer(GstRtspServer.RTSPServer): |
|||
def __init__(self, **properties): |
|||
super(GstServer, self).__init__(**properties) |
|||
self.factory = SensorFactory() |
|||
self.factory.set_shared(True) |
|||
self.get_mount_points().add_factory("/test", self.factory) |
|||
self.attach(None) |
|||
|
|||
|
|||
def run(): |
|||
GObject.threads_init() |
|||
Gst.init(None) |
|||
|
|||
server = GstServer() |
|||
rtsp_port_num = 8554 |
|||
print("\n *** DeepStream: Launched RTSP Streaming at rtsp://localhost:%d/test ***\n\n" % rtsp_port_num) |
|||
loop = GObject.MainLoop() |
|||
loop.run() |
|||
|
|||
|
|||
|
|||
if __name__ == "__main__": |
|||
run() |
|||
|
|||
import subprocess as sp |
|||
import signal |
|||
import numpy as npbool |
|||
#此处换为你自己的地址 |
|||
rtsp_url = 'rtsp://192.168.1.182:8554/video' |
|||
cap = cv2.VideoCapture("rtsp://admin:hk123456@192.168.1.65:554") |
|||
# Get video information |
|||
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)) |
|||
command = ['ffmpeg', |
|||
'-y', |
|||
'-f', 'rawvideo', |
|||
'-vcodec', 'rawvideo', |
|||
'-pix_fmt', 'bgr24', |
|||
'-s', "{}x{}".format(width, height), |
|||
'-r', str(fps), |
|||
'-i', '-', |
|||
# '-c:v', 'libx264', |
|||
'-c:v', 'libx265', |
|||
'-pix_fmt', 'yuv420p', |
|||
'-preset', 'ultrafast', |
|||
'-f', 'rtsp', |
|||
rtsp_url] |
|||
p = sp.Popen(command, stdin=sp.PIPE) |
|||
|
|||
while (cap.isOpened()): |
|||
ret, frame = cap.read() |
|||
frame=cv2.resize(frame,(1080,640)) |
|||
|
|||
if not ret: |
|||
print("Opening camera is failed") |
|||
break |
|||
# frame = 你的图像处理的函数(frame) |
|||
p.stdin.write(frame.tostring()) |
|||
# cv2.imshow('img',frame) |
|||
# if cv2.waitKey(1) == ord('q'): # q to quit |
|||
# raise StopIteration |
|||
|
|||
Loading…
Reference in new issue