yolov5-lite-onnxruntime
yolov5-lite-onnxruntime copied to clipboard
v5lite-s.onnx 模型是怎么从.pt转换来的,请问一下大佬...
我使用官方的 v5lite-s.pt文件 和 官方的 export.py 导出了onnx模型,但是不能在大佬你的代码中使用...
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 : -)