yoloV5-6.1训练的pt模型转换为onnx模型

一、训练模型

yolov5训练自己的数据集-----anaconda安装以及pycharm中环境配置-优快云博客

yolov5训练自己的数据集-----环境搭建，模型准备，训练-优快云博客

二、pt-->onnx

在yolo源文件中有export.py文件

安装onnx库

pip install onnx -i https://pypi.tuna.tsinghua.edu.cn/simple

转换模型，pt模型换成自己的pt文件以及路径

python export.py --weights yolov5s.pt --img 640 --batch 1  # export at 640x640 with batch size 1

转换完成就能看到.onnx文件了。