yolov5 pt文件转换为onnx格式

  1. 安装依赖库pip install onnx coremltools onnx-simplifier

  2. 修改export.py的代码
    在这里插入图片描述
    ①处改成数据源.yaml的路径
    ②处改成模型的权重文件里的.pt的路径

  3. python export.py --include onnx

  4. 即可在根目录找到onxx文件

猜你喜欢

转载自blog.csdn.net/weixin_46504385/article/details/130088774