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.

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

大佬 plate_rec.onnx 输出结果是 [1,21,78]与tensorrt的导出engine,进行推理结果不对 #2

Open
zhuimeng2080 opened this issue Dec 13, 2022 · 3 comments

Comments

@zhuimeng2080
Copy link

请教大佬,pt导出onnx, 输出结果是 [1,21,78]与tensorrt的导出engine,进行推理结果不对

python export.py --weights saved_model/plate_rec.pth --save_path saved_model/plate_rec.onnx --simplify

@we0091234
Copy link
Owner

请教大佬,pt导出onnx, 输出结果是 [1,21,78]与tensorrt的导出engine,进行推理结果不对

python export.py --weights saved_model/plate_rec.pth --save_path saved_model/plate_rec.onnx --simplify

我忘记改了,等会改一下,导出的代码,你重新拉取一下代码

@we0091234
Copy link
Owner

请教大佬,pt导出onnx, 输出结果是 [1,21,78]与tensorrt的导出engine,进行推理结果不对

python export.py --weights saved_model/plate_rec.pth --save_path saved_model/plate_rec.onnx --simplify

重新拉取代码,转onnx的时候加上 --trt

@alice929
Copy link

大佬,请问一下,我通过export.py导出的onnx模型可正确推理出结果,但是通过onnx导出的trt模型,推理结果不正确,请问您有什么办法吗?谢谢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants