Please provide complete information as applicable to your setup.
• Hardware Platform (GPU)
**• DeepStream Version 7.1 **
• TensorRT Version 10.6.0
• NVIDIA GPU Driver Version 560.35.03
• Issue Type FPS drop
• How to reproduce the issue ? I have attached the code
Description
I have connected udpsink after demux, so I can stream rtsp out for each stream separately, it’s working but it leads to severe frame drop for me
demux + udpsink
**PERF: {‘stream0’: 1.0, ‘stream1’: 1.0, ‘stream2’: 1.0, ‘stream3’: 1.0, ‘stream4’: 1.0, ‘stream5’: 1.0}
demux + multifilesink
**PERF: {‘stream0’: 25.2, ‘stream1’: 24.2, ‘stream2’: 24.8, ‘stream3’: 25.0, ‘stream4’: 25.2, ‘stream5’: 25.0}
Goal: Stream RTSP for each input stream individually.
import sys
sys.path.append("../")
import gi
import configparser
import argparse
gi.require_version("Gst", "1.0")
gi.require_version("GstRtspServer", "1.0")
from gi.repository import GstRtspServer
from gi.repository import Gst
from gi.repository import GLib
from ctypes import *
import time
import sys
import os
import math
import platform
from common.bus_call import bus_call
from common.FPS import PERF_DATA
import pyds
import threading
no_display = False
silent = False
file_loop = False
perf_data = None
MAX_DISPLAY_LEN = 64
PGIE_CLASS_ID_VEHICLE = 0
PGIE_CLASS_ID_BICYCLE = 1
PGIE_CLASS_ID_PERSON = 2
PGIE_CLASS_ID_ROADSIGN = 3
MUXER_OUTPUT_WIDTH = 540
MUXER_OUTPUT_HEIGHT = 540
MUXER_BATCH_TIMEOUT_USEC = 4000000
TILED_OUTPUT_WIDTH = 640 #
TILED_OUTPUT_HEIGHT = 360
GST_CAPS_FEATURES_NVMM = "memory:NVMM"
OSD_PROCESS_MODE = 0
OSD_DISPLAY_TEXT = 1
pgie_classes_str = ["person", "helmet"]
def pgie_src_pad_buffer_probe(pad, info, u_data):
frame_number = 0
num_rects = 0
gst_buffer = info.get_buffer()
if not gst_buffer:
print("Unable to get GstBuffer ")
return
batch_meta = pyds.gst_buffer_get_nvds_batch_meta(hash(gst_buffer))
l_frame = batch_meta.frame_meta_list
while l_frame is not None:
try:
frame_meta = pyds.NvDsFrameMeta.cast(l_frame.data)
except StopIteration:
break
frame_number = frame_meta.frame_num
l_obj = frame_meta.obj_meta_list
num_rects = frame_meta.num_obj_meta
while l_obj is not None:
try:
# Casting l_obj.data to pyds.NvDsObjectMeta
obj_meta = pyds.NvDsObjectMeta.cast(l_obj.data)
except StopIteration:
break
try:
l_obj = l_obj.next
except StopIteration:
break
stream_index = "stream{0}".format(frame_meta.pad_index)
global perf_data
perf_data.update_fps(stream_index)
try:
l_frame = l_frame.next
except StopIteration:
break
return Gst.PadProbeReturn.OK
def cb_newpad(decodebin, decoder_src_pad, data):
print("In cb_newpad\n")
caps = decoder_src_pad.get_current_caps()
gststruct = caps.get_structure(0)
gstname = gststruct.get_name()
source_bin = data
features = caps.get_features(0)
print("gstname=", gstname)
if gstname.find("video") != -1:
print("features=", features)
if features.contains("memory:NVMM"):
bin_ghost_pad = source_bin.get_static_pad("src")
if not bin_ghost_pad.set_target(decoder_src_pad):
sys.stderr.write(
"Failed to link decoder src pad to source bin ghost pad\n"
)
else:
sys.stderr.write(
" Error: Decodebin did not pick nvidia decoder plugin.\n")
def decodebin_child_added(child_proxy, Object, name, user_data):
print("Decodebin child added:", name, "\n")
if name.find("decodebin") != -1:
Object.connect("child-added", decodebin_child_added, user_data)
if ts_from_rtsp:
if name.find("source") != -1:
pyds.configure_source_for_ntp_sync(hash(Object))
def create_source_bin(index, uri):
print("Creating source bin")
bin_name = "source-bin-%02d" % index
print(bin_name)
nbin = Gst.Bin.new(bin_name)
if not nbin:
sys.stderr.write(" Unable to create source bin \n")
uri_decode_bin = Gst.ElementFactory.make("uridecodebin", "uri-decode-bin")
if not uri_decode_bin:
sys.stderr.write(" Unable to create uri decode bin \n")
uri_decode_bin.set_property("uri", uri)
uri_decode_bin.connect("pad-added", cb_newpad, nbin)
uri_decode_bin.connect("child-added", decodebin_child_added, nbin)
Gst.Bin.add(nbin, uri_decode_bin)
bin_pad = nbin.add_pad(
Gst.GhostPad.new_no_target(
"src", Gst.PadDirection.SRC))
if not bin_pad:
sys.stderr.write(" Failed to add ghost pad in source bin \n")
return None
return nbin
def make_element(element_name, i):
element = Gst.ElementFactory.make(element_name, element_name)
if not element:
sys.stderr.write(" Unable to create {0}".format(element_name))
element.set_property("name", "{0}-{1}".format(element_name, str(i)))
return element
def create_rtsp_out(rtsp_port_num, udp_ports):
print(f"Initializing RTSP server on port {rtsp_port_num}")
vcodec = "H264"
server = GstRtspServer.RTSPServer.new()
server.props.service = str(rtsp_port_num)
server.attach(None)
for ind, udp_port in enumerate(udp_ports):
factory = GstRtspServer.RTSPMediaFactory.new()
factory.set_launch(
'( udpsrc name=pay0 port=%d buffer-size=524288 caps="application/x-rtp, media=video, clock-rate=90000, encoding-name=%s, payload=96 " )'
% (udp_port, vcodec)
)
factory.set_shared(True)
mount_point = f"/stream{ind}"
print(f"Adding mount point: {mount_point} on UDP port {udp_port}")
server.get_mount_points().add_factory(mount_point, factory)
print(f"DeepStream: Launched RTSP Streaming at rtsp://localhost:{rtsp_port_num}{mount_point}")
def main(args, requested_pgie=None, config=None, disable_probe=False):
input_sources = args
number_sources = len(input_sources)
global perf_data
perf_data = PERF_DATA(number_sources)
rtsp_port_num = 8554
udp_port = 5400
udp_port_list = []
codec = "H264"
bitrate = 4000000
Gst.init(None)
print("Creating Pipeline \n ")
pipeline = Gst.Pipeline()
is_live = False
if not pipeline:
sys.stderr.write(" Unable to create Pipeline \n")
print("Creating streamux \n ")
streammux = Gst.ElementFactory.make("nvstreammux", "Stream-muxer")
if not streammux:
sys.stderr.write(" Unable to create NvStreamMux \n")
pipeline.add(streammux)
for i in range(number_sources):
print("Creating source_bin ", i, " \n ")
uri_name = input_sources[i]
if uri_name.find("rtsp://") == 0:
is_live = True
source_bin = create_source_bin(i, uri_name)
if not source_bin:
sys.stderr.write("Unable to create source bin \n")
pipeline.add(source_bin)
padname = "sink_%u" % i
sinkpad = streammux.get_request_pad(padname)
print("padname {}".format(padname))
if not sinkpad:
sys.stderr.write("Unable to create sink pad bin \n")
srcpad = source_bin.get_static_pad("src")
if not srcpad:
sys.stderr.write("Unable to create src pad bin \n")
srcpad.link(sinkpad)
queue1 = Gst.ElementFactory.make("queue", "queue1")
pipeline.add(queue1)
print("Creating Pgie \n ")
pgie = Gst.ElementFactory.make("nvinfer", "primary-inference")
if not pgie:
sys.stderr.write(" Unable to create pgie \n")
print("Creating nvstreamdemux \n ")
nvstreamdemux = Gst.ElementFactory.make("nvstreamdemux", "nvstreamdemux")
if not nvstreamdemux:
sys.stderr.write(" Unable to create nvstreamdemux \n")
if is_live:
print("Atleast one of the sources is live")
streammux.set_property("live-source", 1)
streammux.set_property("width", 960)
streammux.set_property("height", 540)
streammux.set_property("batch-size", number_sources)
streammux.set_property("batched-push-timeout", 4000000)
if ts_from_rtsp:
streammux.set_property("attach-sys-ts", 0)
pgie.set_property("config-file-path", "dsnvanalytics_pgie_config.txt")
pgie_batch_size = pgie.get_property("batch-size")
if pgie_batch_size != number_sources:
print("WARNING: Overriding infer-config batch-size",pgie_batch_size," with number of sources ",number_sources, " \n")
pgie.set_property("batch-size", number_sources)
print("Adding elements to Pipeline \n")
pipeline.add(pgie)
pipeline.add(nvstreamdemux)
streammux.link(queue1)
queue1.link(pgie)
pgie.link(nvstreamdemux)
print("number_sources-",number_sources)
for i in range(number_sources):
print("i -> {}".format(i))
queue = make_element("queue", i)
pipeline.add(queue)
nvvideoconvert = make_element("nvvideoconvert", i)
pipeline.add(nvvideoconvert)
nvdsosd = make_element("nvdsosd", i)
pipeline.add(nvdsosd)
nvdsosd.set_property("process-mode", OSD_PROCESS_MODE)
nvdsosd.set_property("display-text", OSD_DISPLAY_TEXT)
padname = "src_%u" % i
demuxsrcpad = nvstreamdemux.get_request_pad(padname)
if not demuxsrcpad:
sys.stderr.write("Unable to create demux src pad \n")
queuesinkpad = queue.get_static_pad("sink")
if not queuesinkpad:
sys.stderr.write("Unable to create queue sink pad \n")
demuxsrcpad.link(queuesinkpad)
queue.link(nvvideoconvert)
nvvideoconvert.link(nvdsosd)
nvvidconv_postosd = Gst.ElementFactory.make("nvvideoconvert", "convertor_postosd"+str(i))
if not nvvidconv_postosd:
sys.stderr.write(" Unable to create nvvidconv_postosd \n")
caps = Gst.ElementFactory.make("capsfilter", "filter"+str(i))
caps.set_property("caps", Gst.Caps.from_string("video/x-raw(memory:NVMM),format=I420"))
encoder = Gst.ElementFactory.make("nvv4l2h264enc", "encoder"+str(i))
print("Creating HW H264 Encoder")
if not encoder:
sys.stderr.write(" Unable to create encoder")
rtppay = Gst.ElementFactory.make("rtph264pay", "rtppay"+str(i))
print("Creating H264 rtppay")
if not rtppay:
sys.stderr.write(" Unable to create rtppay")
udp_port += 1
if udp_port not in udp_port_list:
udp_port_list.append(udp_port)
sink = Gst.ElementFactory.make("udpsink", "udpsink"+str(i))
if not sink:
sys.stderr.write(" Unable to create udpsink")
sink.set_property("host", "224.224.255.255")
sink.set_property("port", udp_port)
sink.set_property("async", False)
sink.set_property("sync", 1)
pipeline.add(nvvidconv_postosd)
pipeline.add(caps)
pipeline.add(encoder)
pipeline.add(rtppay)
pipeline.add(sink)
nvdsosd.link(nvvidconv_postosd)
nvvidconv_postosd.link(caps)
caps.link(encoder)
encoder.link(rtppay)
rtppay.link(sink)
print("Linking elements in the Pipeline \n")
loop = GLib.MainLoop()
bus = pipeline.get_bus()
bus.add_signal_watch()
bus.connect("message", bus_call, loop)
pgie_src_pad = pgie.get_static_pad("src")
if not pgie_src_pad:
sys.stderr.write(" Unable to get src pad \n")
else:
pgie_src_pad.add_probe(Gst.PadProbeType.BUFFER, pgie_src_pad_buffer_probe, 0)
GLib.timeout_add(5000, perf_data.perf_print_callback)
print("Now playing...")
for i, source in enumerate(input_sources):
print(i, ": ", source)
print("Starting pipeline \n")
print('rtsp_port_num',rtsp_port_num)
print('udp_port_list',udp_port_list)
print('number_sources',number_sources)
create_rtsp_out(rtsp_port_num, udp_port_list)
pipeline.set_state(Gst.State.PLAYING)
try:
loop.run()
except:
pass
print("Exiting app\n")
pipeline.set_state(Gst.State.NULL)
def parse_args():
parser = argparse.ArgumentParser(description='RTSP Output Sample Application Help ')
parser.add_argument("-i", "--input",
help="Path to input H264 elementry stream",
nargs="+", default=["a"], required=True)
parser.add_argument("-g", "--gie", default="nvinfer",
help="choose GPU inference engine type nvinfer or nvinferserver , default=nvinfer", choices=['nvinfer','nvinferserver'])
parser.add_argument("-c", "--codec", default="H264",
help="RTSP Streaming Codec H264/H265 , default=H264", choices=['H264','H265'])
parser.add_argument("-b", "--bitrate", default=4000000,
help="Set the encoding bitrate ", type=int)
parser.add_argument("--rtsp-ts", action="store_true", default=False, dest='rtsp_ts', help="Attach NTP timestamp from RTSP source",
)
# Check input arguments
if len(sys.argv)==1:
parser.print_help(sys.stderr)
sys.exit(1)
args = parser.parse_args()
global codec
global bitrate
global stream_path
global gie
global ts_from_rtsp
gie = args.gie
codec = args.codec
bitrate = args.bitrate
stream_path = args.input
ts_from_rtsp = args.rtsp_ts
return stream_path
if __name__ == "__main__":
stream_paths = parse_args()
sys.exit(main(stream_paths))