从 yolov4 或 yolov5 转换为 coreml答案

【问题标题】：converting from yolov4 or yolov5 to coreml从 yolov4 或 yolov5 转换为 coreml
【发布时间】：2020-12-21 21:01:15
【问题描述】：

将 yolov4 或 yolov5 模型转换为 coreml 的典型过程是什么。我看到了从 Pytorch 到 Onnx 到 CoreML 的东西。我正在使用 turicreate 并陷入 yolov2 土地。有人在 v4 或 v5 上试过吗？

【问题讨论】：

也卡在这里...我可以转换为 .h5，但在转换为 coreml 时面临大量问题。你能成功地将yolov2转换成coreml吗？
我使用 Turicreate 来训练一个 yolov2 模型并生成一个 coreml 模型，但我们现在已经达到了 yolov5。 Turicreate 在新功能或新模型支持方面似乎已经死了。所以我想我们需要弄清楚如何将更高的 yolo 模型移植到 coreml..
这个文档我想我们可以使用 coreml 工具 coremltools.readme.io/docs
是的，您可以尝试在 tensorflow、keras 或 pytorch 中进行训练，然后进行转换。但这似乎并不容易。或者你可以问他们，他们有一个用于 v3 和 v4 和 (v5) 的转换管道，但并不便宜：github.com/ultralytics/yolov5
是的，我们想出了使它工作的步骤。我意识到这个 ultralytics 遗漏了很多步骤。在我看来有点阴暗。现在我们正在尝试弄清楚如何将我们在 yolov2 中的 swift 框架更改为在 yolov5 中工作。一旦我解决了这个问题，我将发布完整的解决方案。

【解决方案1】：

将 YOLOv5 模型转换为 CoreML 格式：

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

【讨论】：