使用 openCV 从 ONNX 加载网络时遇到问题（python/pytorch）答案

【问题标题】：Trouble using openCV to load a net from ONNX (python/pytorch)使用 openCV 从 ONNX 加载网络时遇到问题（python/pytorch）
【发布时间】：2019-04-12 23:05:01
【问题描述】：

我正在尝试将经过训练的 .onnx 模型（来自神经风格转移算法）加载到 cv2 中。我看到有一个

cv.dnn.readNetFromONNX()

函数，但是cv2中没有这个函数。我似乎无法将 opencv 作为 cv 导入或加载，因此似乎无法将我的模型加载到 cv2 中。有谁知道解决办法吗？

我基本上已经用 https://github.com/pytorch/examples/blob/master/fast_neural_style/neural_style/neural_style.py#L122-L150 此脚本，并通过添加导出 onnx 模型

torch.onnx.export(style_model, dummy_input, "chipsoft_mod.onnx", verbose=True)

现在我想通过 cv2 阅读器运行经过训练的模型，但我失败了。

【问题讨论】：

也遇到过这个问题，又一个使用/学习C++的好理由

标签： python opencv pytorch onnx

【解决方案1】：

将您的 opencv 更新到较新的版本。它应该会有所帮助。

pip install opencv-python==4.1.0.25

【讨论】：