we0091234 / crnn_plate_recognition

crnn chinese_plate_recognition
307 stars 61 forks source link

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

Open zhuimeng2080 opened 1 year ago

zhuimeng2080 commented 1 year ago

请教大佬,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 commented 1 year ago

请教大佬,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 commented 1 year ago

请教大佬,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 commented 1 year ago

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