yolov5 pt文件转换为onnx格式

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

  2. 修改export.py的代码
    yolov5 pt文件转换为onnx格式_第1张图片
    ①处改成数据源.yaml的路径
    ②处改成模型的权重文件里的.pt的路径

  3. python export.py --include onnx

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

你可能感兴趣的:(yolo学习,yolo)