onnx - 将yolov5模型转换为ONNX并在c++接口上运行

Question

我有 yolo 模型，yolov5s.yaml并且我已将我的权重文件保存为best.pt. 现在想将 yolo 模型转换为 ONNX 并在 c++ 接口上运行。我已经通过了这个https://github.com/ultralytics/yolov5/issues/251但无法做到。任何帮助或有用的资源都会有所帮助。谢谢

Answer 1

我尝试成功运行第一个链接中提供的示例，语句为： git clone github.com/ultralytics/yolov5 # clone repo cd yolov5 pip install -r requirements.txt # base requirements pip install coremltools>=4.1 onnx>=1.9 .0 scikit-learn==0.19.2 # 导出要求 python models/export.py --weights yolov5s.pt --img 640 --batch 1 # 以 640x640 导出，批量为 1

说明导出成功。上例中的权重文件是 yolov5s.pt。通过使用 best.pt 文件运行这些语句，您得到了什么样的错误？

Answer 2

我可以成功运行https://github.com/ultralytics/yolov5/issues/251中提到的示例。我的手电筒版本是 1.8.0。

可能您的权重文件 best.pt 具有无效值。您是否曾尝试使用上述链接中的权重文件成功导出模块？

顺便说一句：您可以在https://github.com/microsoft/onnxruntime/tree/master/samples/c_cxx中找到使用 ONNXRuntime C++ api 运行它的示例。