23 How to change python sample apps from display to output file or fakesink for the users who do not have monitor in their device, the patch is based on test1 sample.
Usage: python3 deepstream_test_1.py <media file or uri> <sink type: 1-filesink; 2-fakesink; 3-display sink>
nvidia@ubuntu:/opt/nvidia/deepstream/deepstream/sources/deepstream_python_apps/apps/deepstream-test1$ diff -Naur deepstream_test_1.py.orig deepstream_test_1.py
--- deepstream_test_1.py.orig 2022-08-15 20:12:39.809775283 +0800
+++ deepstream_test_1.py 2022-08-15 22:06:27.052250778 +0800
@@ -123,8 +123,8 @@
def main(args):
# Check input arguments
- if len(args) != 2:
- sys.stderr.write("usage: %s <media file or uri>\n" % args[0])
+ if len(args) != 3:
+ sys.stderr.write("usage: %s <media file or uri> <sink type: 1-filesink; 2-fakesink; 3-display sink>\n" % args[0])
sys.exit(1)
# Standard GStreamer initialization
@@ -179,14 +179,46 @@
if not nvosd:
sys.stderr.write(" Unable to create nvosd \n")
- # Finally render the osd output
- if is_aarch64():
- transform = Gst.ElementFactory.make("nvegltransform", "nvegl-transform")
-
- print("Creating EGLSink \n")
- sink = Gst.ElementFactory.make("nveglglessink", "nvvideo-renderer")
- if not sink:
- sys.stderr.write(" Unable to create egl sink \n")
+ if args[2] == '1':
+
+ nvvidconv1 = Gst.ElementFactory.make ("nvvideoconvert", "nvvid-converter1")
+ if not nvvidconv1:
+ sys.stderr.write("Unable to create nvvidconv1")
+ capfilt = Gst.ElementFactory.make ("capsfilter", "nvvideo-caps")
+ if not capfilt:
+ sys.stderr.write("Unable to create capfilt")
+ caps = Gst.caps_from_string ('video/x-raw(memory:NVMM), format=I420')
+# feature = gst_caps_features_new ("memory:NVMM", NULL)
+# gst_caps_set_features (caps, 0, feature)
+ capfilt.set_property('caps', caps)
+ print("Creating nvv4l2h264enc \n")
+ nvh264enc = Gst.ElementFactory.make ("nvv4l2h264enc" ,"nvvideo-h264enc")
+ if not nvh264enc:
+ sys.stderr.write("Unable to create nvh264enc")
+ print("Creating filesink \n")
+ sink = Gst.ElementFactory.make ("filesink", "nvvideo-renderer")
+ sink.set_property('location', './out.h264')
+ if not sink:
+ sys.stderr.write("Unable to create filesink")
+
+ elif args[2] == '2':
+
+ print("Creating fakesink \n")
+ sink = Gst.ElementFactory.make ("fakesink", "fake-renderer")
+ if not sink:
+ sys.stderr.write("Unable to create fakesink")
+
+ elif args[2] == '3':
+
+ print("Creating EGLSink \n")
+ sink = Gst.ElementFactory.make("nveglglessink", "nvvideo-renderer")
+ if not sink:
+ sys.stderr.write(" Unable to create egl sink \n")
+ if is_aarch64():
+ transform = Gst.ElementFactory.make("nvegltransform", "nvegl-transform")
+ if not transform:
+ sys.stderr.write(" Unable to create egl transform \n")
print("Playing file %s " %args[1])
source.set_property('location', args[1])
@@ -204,9 +236,17 @@
pipeline.add(pgie)
pipeline.add(nvvidconv)
pipeline.add(nvosd)
- pipeline.add(sink)
- if is_aarch64():
- pipeline.add(transform)
+ if args[2] == '1':
+ pipeline.add(nvvidconv1)
+ pipeline.add(capfilt)
+ pipeline.add(nvh264enc)
+ pipeline.add(sink)
+ elif args[2] == '2':
+ pipeline.add(sink)
+ elif args[2] == '3':
+ pipeline.add(sink)
+ if is_aarch64():
+ pipeline.add(transform)
# we link the elements together
# file-source -> h264-parser -> nvh264-decoder ->
@@ -225,11 +265,19 @@
streammux.link(pgie)
pgie.link(nvvidconv)
nvvidconv.link(nvosd)
- if is_aarch64():
- nvosd.link(transform)
- transform.link(sink)
- else:
+ if args[2] == '1':
+ nvosd.link(nvvidconv1)
+ nvvidconv1.link(capfilt)
+ capfilt.link(nvh264enc)
+ nvh264enc.link(sink)
+ elif args[2] == '2':
nvosd.link(sink)
+ elif args[2] == '3':
+ if is_aarch64():
+ nvosd.link(transform)
+ transform.link(sink)
+ else:
+ nvosd.link(sink)
# create an event loop and feed gstreamer bus mesages to it