3

我正在尝试将经过训练的 .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 阅读器运行经过训练的模型,但我失败了。

4

1 回答 1

1

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

pip install opencv-python==4.1.0.25
于 2019-06-13T14:56:24.407 回答