我的yolo模型是yolov5s.yaml
,我的权重文件保存为best.pt
。现在想要将yolo模型转换为ONNX并在c++接口上运行。我已经经历过了https://github.com/ultralytics/yolov5/issues/251但做不到。任何帮助或有用的资源都会有所帮助。感谢
我可以运行https://github.com/ultralytics/yolov5/issues/251成功地我的torch版本是1.8.0。
可能您的权重文件best.pt的值无效。您是否曾尝试使用上面链接中的权重文件成功导出模块?
BTW:你可以在https://github.com/microsoft/onnxruntime/tree/master/samples/c_cxx使用ONNXRuntime C++api来运行它。
我尝试成功运行第一个链接中提供的示例,语句为:git clone github.com/ultralytics/yolov5#clone repocd yolov5pip install-r requirements.txt#基本要求pip安装coremltools>4.1 onnx>1.9.0 scikit learn==0.19.2#导出要求python models/export.py--weights yolov5s.pt--img 640--批次1#以640x640导出,批次大小为1
它显示导出成功。上面例子中的weight文件是yolov5s.pt。使用best.pt文件运行这些语句会产生什么样的错误?