PyTorch 到 onnx 并与 opencv-dnn 一起使用?

Posted

技术标签:

【中文标题】PyTorch 到 onnx 并与 opencv-dnn 一起使用?【英文标题】:PyTorch to onnx and use with opencv-dnn? 【发布时间】:2021-10-16 15:33:20 【问题描述】:

我想在 C++ 中使用 opencv dnn 运行 yolov5,为此,我已将 PyTorch 模型转换为 onnx,来自 link 但是那个 onnx 不适用于 opencv dnn 模块。 任何帮助将不胜感激。

【问题讨论】:

万一,你找到解决办法了吗? 【参考方案1】:

我已经按照下面的链接成功地用 C++ 运行了 yolov5

Link-1

Link-2

【讨论】:

以上是关于PyTorch 到 onnx 并与 opencv-dnn 一起使用?的主要内容,如果未能解决你的问题,请参考以下文章

对象检测模型 (PyTorch) 到 ONNX:ONNX 推理的空输出

pytorch 转tensorflow注意

PyTorch模型导出到ONNX文件示例(LeNet-5)

无法将 PyTorch 模型导出到 ONNX

PyTorch 1.0 中文官方教程:使用ONNX将模型从PyTorch传输到Caffe2和移动端

Pytorch 到 ONNX 导出功能失败并导致遗留功能错误