PaddleDetection
PaddleDetection copied to clipboard
关于用RT-DETR训练的模型转成onnx后的推理问题
问题确认 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,请问有对应的推理代码吗?(我目前调试的那个推理代码有点问题)
转成trt后用这里的代码
https://github.com/lyuwenyu/RT-DETR/blob/main/benchmark/trtinfer.py