主要环境要求:
python >= 3.8
pytorch = 1.7.0
onnx = 1.10.1
onnx-tf = 1.9.0
此处以转yolov5s.pt模型为例,转换成onnx模型命令如下:
python models/export.py --weights yolov5s.pt --img 640 --batch 1
若为自己训练的数据集模型为例,命令可参考如下:
python models/export.py --weights runs/train/exp/weights/best.pt --img 640 --batch 1
但是第一次转换没能成功,报错如下:
ImportError: /usr/lib/x86_64-linux-gnu/libstdtt.so.6: version ‘GLIBDKX_3.4.22' not found (requined by /hone/xinxiB11/anaconda3/envs/yolov5/lib/python3.6/site-packages/scipy/fft/_pocketfft/pypocketfft.cpython-361-x86_64-linux-gnu.so)
根据搜索可知,为默认路径下libstdc++.so.6缺少GLIBCXX_3.4.22,
先使用指令查看当下有哪些版本:
strings /usr/lib/x86_64-linux-gnu/libstdc++.so.6 | grep GLIBCXX
结果为: