Cloud Message (Kafka) Configurations for YOLO models

Hi!

I have managed to run deepstream-test5 and YOLOV7 models with DeepStream seperately. In deepstream-test5, I configured a sink component that sinks the objects to the Kafka and it works, but the same configuration didn’t work with the YOLOV7 model. Here is the sink component that I use:

[sink1]
enable=1
type=6
msg-conv-config=dstest5_msgconv_sample_config.txt
msg-conv-payload-type=1
msg-conv-msg2p-new-api=0
msg-conv-frame-interval=30
msg-broker-proto-lib=/opt/nvidia/deepstream/deepstream/lib/libnvds_kafka_proto.so
msg-broker-conn-str=172.16.44.101;32438;my-topic
topic=my-topic
msg-broker-config=/opt/nvidia/deepstream/deepstream/sources/libs/kafka_protocol_adaptor/cfg_kafka.txt

My aim is to generate Kafka messages when the YOLOV7 model detects a specific object. What are the requirements for this purpose? Should I write a custom code using SDK to achieve this, or is it configurable via pipeline/inference configurations?

Any kind of help is appreciated!

•Hardware Platform: Docker Container (nvcr.io/nvidia/deepstream:6.4-gc-triton-devel) / PC (x86_64) / NVIDIA GeForce GTX 1650
•DeepStream Version: 6.4
•TensorRT Version: 8.6.1.6-1+cuda12.0
•NVIDIA GPU Driver Version: 535.161.08

please make sure yolov7 can work well. for example, if using filesink, you can see the bbox on the video. if yolo7 works well in deeptream-test5, you can add some logs in before generate_event_msg_meta of bbox_generated_probe_after_analytics in deepstream_test5_app_main.c to check if generate_event_msg_meta is called.

I removed secondary GIEs and configured the primary GIE as below, and it worked:

[primary-gie]
enable=1
gpu-id=0
batch-size=1
bbox-border-color0=1;0;0;1
bbox-border-color1=0;1;1;1
bbox-border-color2=0;0;1;1
bbox-border-color3=0;1;0;1
interval=0
gie-unique-id=1
nvbuf-memory-type=0
labelfile-path=/opt/nvidia/deepstream/deepstream-6.4/sources/apps/sample_apps/deepstream-test5/labels.txt
config-file=/opt/nvidia/deepstream/deepstream-6.4/sources/apps/sample_apps/deepstream-test5/config_infer_primary_yoloV7.txt

ps. Place labels.txt and config_infer_primary_yoloV7.txt to the specified paths.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.