2

有人可以帮我将 YOLOv5 PyTorch 模型转换为 ONNX 或 TensorFlow 格式，以便能够将其与 OpenCV C++ 推理一起使用吗？

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

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

4

2 回答 2

1

OpenCV 4.5.4 可以读取的 pt 模型到 onnx 如下：

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

于 2021-12-03T16:52:36.097 回答
1

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

原始线程

好消息 ！您的原始问题现在可能已由@SamFC10 在 PR #4833中修复 ✅。此 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)
  • 笔记本– 查看更新的笔记本 在 Colab 中打开 在 Kaggle 中打开。Colab
  • Docker——sudo docker pull ultralytics/yolov5:latest更新你的镜像。

感谢您发现此问题并告知我们该问题。如果此更新为您解决了问题，请告诉我们，如果您发现任何其他问题或想到的功能请求，请随时通知我们。使用 YOLOv5 进行愉快的训练！

于 2021-10-14T11:55:29.077 回答