Home

Awesome

DeepStream-Yolo-Seg

NVIDIA DeepStream SDK application for YOLO-Seg models


YOLO objetct detection models and other infos: https://github.com/marcoslucianops/DeepStream-Yolo


Getting started

Supported models

Instructions

1. Download the DeepStream-Yolo-Seg repo

git clone https://github.com/marcoslucianops/DeepStream-Yolo-Seg.git
cd DeepStream-Yolo-Seg

2. Compile the libs

3. Run

deepstream-app -c deepstream_app_config.txt

NOTE: The TensorRT engine file may take a very long time to generate (sometimes more than 10 minutes).

NMS configuration

For now, the NMS is configured in the ONNX exporter file.

NOTE: Make sure to set cluster-mode=4 in the config_infer file.

Detection threshold configuration

The minimum detection confidence threshold is configured in the ONNX exporter file. The pre-cluster-threshold should be >= the value used in the ONNX model.

[class-attrs-all]
pre-cluster-threshold=0.25
topk=100

My projects: https://www.youtube.com/MarcosLucianoTV