crnn_plate_recognition icon indicating copy to clipboard operation
crnn_plate_recognition copied to clipboard

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

Open zhuimeng2080 opened this issue 2 years ago • 3 comments

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

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

zhuimeng2080 avatar Dec 13 '22 03:12 zhuimeng2080

请教大佬,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 avatar Dec 13 '22 05:12 we0091234

请教大佬,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

we0091234 avatar Dec 13 '22 05:12 we0091234

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

alice929 avatar Mar 10 '23 10:03 alice929