我正在尝试将经过训练的.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阅读器运行经过训练的模型，但是我失败了。
最佳答案
将您的opencv更新到新版本。应该会有所帮助。
pip install opencv-python==4.1.0.25
关于python - 使用openCV从ONNX加载网络时遇到问题(python/pytorch)，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/53224685/