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 推理的空输出