使用 YOLOv5 提供的 export.py 将 yolov5s.pt 转换为 ONNX

使用 YOLOv5 提供的 export.py 可以将经过训练的 YOLOv5 模型转换为 ONNX 格式。您可以按照以下步骤进行操作：

下载 YOLOv5 代码和模型权重文件，具体方法请参考 YOLOv5 官方网站。

进入 YOLOv5 代码的根目录，并执行以下命令安装依赖：

pip install -r requirements.txt
执行以下命令将 YOLOv5 模型转换为 ONNX 格式：

python export.py --weights path/to/yolov5s.pt --img 640 --batch 1 --simplify --dynamic --include all
其中，--weights 参数指定模型权重文件的路径，--img 参数指定输入图像的大小，--batch 参数指定批量大小，--simplify 参数表示简化模型，--dynamic 参数表示导出动态形状的模型，--include all 表示导出所有输出层。

执行命令后，将生成一个名为 yolov5s.onnx 的 ONNX 模型文件，可以使用 TensorFlow、PyTorch 或 ONNX Runtime 等框架进行推理。