mmdeploy icon indicating copy to clipboard operation
mmdeploy copied to clipboard

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

Open Hou-MY opened this issue 1 year ago • 2 comments

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.

Hou-MY avatar Aug 15 '22 07:08 Hou-MY

Hi, you can refer to the PR.

AllentDan avatar Aug 15 '22 09:08 AllentDan

Hi, you can refer to the PR.

thanks,it work

machine52vision avatar Sep 03 '22 04:09 machine52vision