我正试图将一个经过训练的.onx模型(来自神经风格的传递算法(加载到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