yolov5-lite-onnxruntime icon indicating copy to clipboard operation
yolov5-lite-onnxruntime copied to clipboard

v5lite-s.onnx 模型是怎么从.pt转换来的,请问一下大佬...

Open Singformeasong opened this issue 3 years ago • 2 comments

我使用官方的 v5lite-s.pt文件 和 官方的 export.py 导出了onnx模型,但是不能在大佬你的代码中使用...

Singformeasong avatar Feb 25 '22 10:02 Singformeasong

Hello! Though I'm not this repo's contributor, I can provide some suggestions that may assist u.

Please check that at least the 5 args below are included when u use export.py in ppogg's YOLOV5-Lite repo: --weights xxxx --img-size xxxx --batch-size xxxx --device xxxx --grid (I forgot this one hince the .py project falled to run)

Besides, u might also need to check that the version of onnx module is 1.8.1 (just pip install onnx==1.8.1).

Hope it be helpful : -)

defzhangaa avatar Jun 12 '24 04:06 defzhangaa