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.

Sign up for GitHub

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

Jump to bottom

How can i add a new custom op for TensorRT in mmdeploy? #901

Closed
Hou-MY opened this issue Aug 15, 2022 · 2 comments
Closed

How can i add a new custom op for TensorRT in mmdeploy? #901

Hou-MY opened this issue Aug 15, 2022 · 2 comments
Assignees
@AllentDan

Comments

@Hou-MY
Copy link

Hou-MY commented Aug 15, 2022

Is there any example or tutorial？ I just saw the procedures of adding custom operators for ONNX Runtime in MMDeploy. I want to implement soft_nms op in TensorRT .Because the mAP of trt is lower than pth.
@AllentDan AllentDan self-assigned this Aug 15, 2022
@AllentDan
Copy link
Member

Hi, you can refer to the PR.

@machine52vision
Copy link

Hi, you can refer to the PR.

thanks,it work

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@AllentDan AllentDan

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@AllentDan @machine52vision @Hou-MY