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

Question

我正在尝试将经过训练的.onnx模型（来自神经样式传输算法）加载到cv2中。我已经看到有一个

package auth

import (
    "fmt"
    "path/to/config"
    log "github.com/sirupsen/logrus"
    "net/http"
)
func LoginHandler(w http.ResponseWriter, r *http.Request) {

    username := r.FormValue("username")
    password := r.FormValue("password")

    status, userName, mail, _ := auth(username, password)
    //some code goes here

}

功能，但是cv2中没有这样的功能。我似乎无法将opencv导入或加载为cv，因此似乎无法在cv2中加载我的模型。有人知道解决方案吗？

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

来导出onnx模型

cv.dnn.readNetFromONNX() 

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

Answer 1

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

pip install opencv-python==4.1.0.25