YOLOV5模型转换(pt转onnx)

1.安装依赖库

pip install onnx coremltools onnx-simplifier

2.导出 onnx

python models/export.py --weights best.pt --img 640 --batch 1

3.用 onnx-simplifer 简化模型

python -m onnxsim best.onnx best-sim.onnx

你可能感兴趣的:(python,神经网络,深度学习,计算机视觉)