ONNX: export failure 2.2s: DLL load failed while importing onnx_cpp2py_export: (DLL)

背景

将通过yolov5s生成的模型best.pt转换为best.onnx遇到这个问题, 原因是包的版本冲突, 各个包的版本及转换命令见 解决方法。

解决方法

#安装包
pip install onnx==1.15.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install protobuf==5.27.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install onnxconverter-common==1.13.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

#转换为onnx
# export.py为yolov5中自带的脚本,  --weight为自己的模型所在的位置
python export.py --weights runs/train/exp2/weights/best.pt --include onnx

• export.py所在位置
  
• 自己的模型(best.pt)所在的位置
  

注意

只使用 python export.py --weights runs/train/exp2/weights/best.pt 可能会出现只生成 .torchscript 文件, 可以在命令后加上 --include onnx ,完整命令见 解决办法。