marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
MIT License
1.38k stars 343 forks source link

Sending Bounding Box info through UDP connection #518

Open jsaffyvtti opened 4 months ago

jsaffyvtti commented 4 months ago

Hi I am using deepstream yolo for traffic monitoring purposes and am trying to send the bounding box info through a udp connection. I am familiar with deep learning and computer vision models specifically but am new to this low level cuda programming and video streaming and was having trouble setting up a pipeline with gi and pyds.

So I tried going into the nvdsparsebbox_Yolo_cuda.cu file and set up a UDP server but am unable to import any c++ code into the decodeTensorYoloCuda function, I am unfamiliar with cuda programming but I see now that it has to do with global and device functions although I'm unsure of how to fix that.

I then found the deepstream_app.c code in the opt/deepstream/... folder and tried editing that. I was able to get it to compile and run but it seems it didn't do any of the changes I made to the code which makes me think I'm editing the wrong file.

If anyone has any insight on how I may be able to fix my issues, I'd appreciate the help!

Thanks

Josh

marcoslucianops commented 3 months ago

You need to use the apps provided by NVIDIA (python or C/C++) and add you logic there. You can use this plugin with any DeepStream code.