-
Notifications
You must be signed in to change notification settings - Fork 19
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Deepstream integration for Yolo-pose #15
Comments
can you tell me, what version of tensorrt you using? i am having issue on creating tensorrt engine. |
I am also interested in deploying the key point model with deepstream. Any update on this? |
Hi, the tensorrt version is 8.5.2.2 but I had it also working in tensorrt 8.4.1. One thing to note, depending on the cuda/tensorrt version, the node numbers in the onnx file will change in add_custom_yolo_op.py. Please use netron to visualize the onnx model and use the correct numbers for inputs. |
No update yet. I am keeping an eye on this repo also to see if there is any update, https://github.com/marcoslucianops/DeepStream-Yolo.git. |
I am interested in learning how to develop the neccessary plugins ando/or parsers for yolo pose deepstream. How could I start ? is there any documentation for that? btw, @MarcoPrassel , could you share your progress with the code? |
Hi |
Hi,
Great work with the repo, Thank you for that. I was able to generate a yolo pose fp16 engine using this repo and get it to work on my GPU (RTX 2070).
Have you tried or are you aware of using the generated engine with nvinfer plugin of deepstream? I want to use it in deepstream 6.1.1 (with Cuda 11.7).
The text was updated successfully, but these errors were encountered: