You're reading an pre-release version of this documentation.
For the latest stable release version, please have a look at master.

Dismiss

PyTorch 转 ONNX 示例

YOLOv5 模型转 ONNX

本示例展示 yolov5s.pt 转 ONNX 的方法。

环境搭建

mamba create -n conversion python==3.10
mamba activate conversion
git clone https://github.com/ultralytics/yolov5.git
cd yolov5
pip install -r requirements.txt

转换命令

python export.py --weights yolov5s.pt --include onnx

脚本会自动从官方地址下载权重，执行完成后终端输出：

ONNX: export success  19.8s, saved as yolov5s.onnx (28.0 MB)

yolov5s.onnx 生成在当前目录下。

---

YOLOv8 模型转 ONNX

本示例展示 yolov8s.pt 转 ONNX 的方法。

环境搭建

mamba create -n conversion python==3.10
mamba activate conversion
git clone https://github.com/Pertical/YOLOv8.git
cd YOLOv8
pip install ultralytics

转换命令

yolo export model=yolov8s.pt format=onnx

脚本会自动从官方地址下载权重，执行完成后终端输出：

ONNX: export success  15.0s, saved as 'yolov8s.onnx' (42.8 MB)

yolov8s.onnx 生成在当前目录下。

---

作者：ougaoliang