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