如何将 YOLOv5 PyTorch 模型转换为 OpenCV DNN 兼容格式答案

【问题标题】：How to convert YOLOv5 PyTorch model to OpenCV DNN compatible format如何将 YOLOv5 PyTorch 模型转换为 OpenCV DNN 兼容格式
【发布时间】：2021-10-31 22:45:20
【问题描述】：

谁能帮我将 YOLOv5 PyTorch 模型转换为 ONNX 或 TensorFlow 格式，以便能够将其用于 OpenCV C++ 推理？

我使用本教程使用 colab 训练模型：https://blog.roboflow.com/how-to-train-yolov5-on-a-custom-dataset/

也许在 colab 中有一种转换方式，或者我应该使用任何其他工具？

【问题讨论】：

标签： c++ opencv yolov5 dotnetnuke roboflow

【解决方案1】：

以下是最近 Ultralytics 开发人员（YoloV5 创建者）关于 OpenCV 和 YoloV5 兼容性的回复。其实我也在找正确的实现教程。

?Original Thread

好消息?！您的原始问题现在可能已在 PR #4833 中修复✅ 通过@SamFC10。此 PR 实施架构更新以允许 ONNX 导出的 YOLOv5 模型与 OpenCV DNN 一起使用。

要接收此更新：

Git - git pull 来自 your yolov5/ 目录或 git clone https://github.com/ultralytics/yolov5

PyTorch Hub – 使用 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True) 强制重新加载

Notebooks – 查看更新的笔记本 Open In Colab Open In Kaggle。 Colab

Docker – sudo docker pull ultralytics/yolov5:latest 更新您的图片。

感谢您发现此问题并告知我们该问题。请让我们知道此更新是否为您解决了问题，并感觉免费告知我们您发现的任何其他问题或功能请求我想到了。 YOLOv5 的快乐训练?！

【讨论】：

【解决方案2】：

OpenCV 4.5.4能够读取的onnx的pt模型如下：

python3 export.py --weights best.pt --include onnx --simplify

【讨论】：