Skip to content
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 compatible yolov8-seg Custom mask and bbox plugin #215

Open
neeraj-satyaki opened this issue May 6, 2024 · 1 comment
Open

Comments

@neeraj-satyaki
Copy link

Hi triple-Mu,
I am really impressed by your work, its a lot of contribution for Yolov8.
I would request one more thing,

I am trying to export the model from nms_seg_plugin branch export code for some reason I could not export it.
Anyway I was able to do it with yolo cli, it looks like, tensorrt engine has different set of output layers. like
User
0 INPUT kFLOAT images 3x640x640
1 OUTPUT kFLOAT output1 32x160x160
2 OUTPUT kFLOAT output0 38x8400
Which is handled using CV2.
Could you give an idea, how we can implement this in deepstream custom parser plugin without CV2. Or directly implement it.

@triple-Mu
Copy link
Owner

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants