Turoad
Code for custom ops for conversion to onnx and trt.
Hello @Turoad . Thanks for sharing your code and for the great work you have accomplished. In another issue, you mentioned that you have written custom operators for roialign, possibly grid_sample, etc to convert the torch model to onnx. Could you please share your code for the same? Thanks in advance!
Hi @MySuperSoul , following up on my issue here. Could you share your code and procedure of using custom ops to convert the model to onnx? Thanks!
@Turoad @MySuperSoul Thanks for letting me know. I'm really looking forward for this update.
You can refer to my work from https://github.com/xuanandsix/CLRNet-onnxruntime-and-tensorrt-demo An initial version code can implement onnxruntime inference and I will still try optimize it further. I also try to make a trt demo.
