Segmentation with deepstream using Python

Please provide complete information as applicable to your setup.

• Hardware Platform (Jetson / GPU): RTX 2080
• DeepStream Version: 5.0
• JetPack Version (valid for Jetson only)
• TensorRT Version: 7.0
• NVIDIA GPU Driver Version (valid for GPU only): 410.xx

Hi, Im currently working on segmentation problem using deepstream with python, I have also read over the deepstream-python-apps for samples, but still struggle deploying a pretrained model in Pytorch to deepstream. Is there any reference or code snippets that I could get along with, especially working with segmentation task. I have found a repos seems to work with segmentation problem using deepstream. As far as I understand, the flow is almost the same with the detection task, so what distinguishes those 2 tasks, is it the metadata, cause the manual said that the input to nvsegvisual plugin is NvDsInferSegmentationMeta.
Link to repos:
Link to manual:

Hi @redalert1013,
DS 5.0 GA will support a segmentation network - mask-rcnn along with some enhancement for the segmentation network, so could you wait for the DS5.0GA that is coming very soon (in 1~2 weeks).


1 Like