PaddleDetection icon indicating copy to clipboard operation
PaddleDetection copied to clipboard

关于用RT-DETR训练的模型转成onnx后的推理问题

Open wmjlincy opened this issue 2 years ago • 1 comments

问题确认 Search before asking

  • [X] 我已经搜索过问题,但是没有找到解答。I have searched the question and found no related answer.

请提出你的问题 Please ask your question

您好,我使用FocalNet-L作为RT-DETR的backbone,训练结束后,使用tools/infer.py进行推理,效果还是可以的,现在想将模型转成onnx,请问有对应的推理代码吗?(我目前调试的那个推理代码有点问题)

wmjlincy avatar Aug 04 '23 01:08 wmjlincy

转成trt后用这里的代码

https://github.com/lyuwenyu/RT-DETR/blob/main/benchmark/trtinfer.py

lyuwenyu avatar Aug 10 '23 09:08 lyuwenyu