Need help with custom model deepstream-app

Please provide complete information as applicable to your setup.

• Jetson Nano 4GB
• DeepStream 5.1
• JetPack 4.5.1

Hi,
I’m new to deepstream community yet i’m enjoying the challenge. I have a question. I have successfully implemented the deepstream-app in the samples folder. It’s running 23 FPS. As you know it has 80 classes in it. Can someone show me the way of how to make it work for only one or two classes(such as person, car)? Custom_YOLO_Model_in_the_DeepStream_YOLO_App.pdf (103.1 KB)
I’ve seen this document but could not resolve the issue.
Thanks.

Hi,

Do you want to use a custom model or just want to filter out some output labels?
For the later, you can check if filter-out-class-ids or operate-on-class-ids can meet your requirement.

https://docs.nvidia.com/metropolis/deepstream/dev-guide/text/DS_plugin_gst-nvinfer.html#gst-nvinfer-file-configuration-specifications

Thanks.

Hi AastaLLL,

I think i would like to learn how to use a custom model. How should i go about it?

Thanks.

Hi,

Please find the guidance in the below document:
https://docs.nvidia.com/metropolis/deepstream/dev-guide/text/DS_using_custom_model.html

In general, it can work by updating the model and data path directly.
Thanks.

Thanks for your help @AastaLLL