使用 Movidius NCS2 的 OpenCV 在使用暗网神经网络时会出现错误

Question

我正在尝试使用英特尔 Movidius 神经电脑棒 2 执行 OCR。我正在使用的 OCR 网络基于 YOLO，因此将图形结构保存为.cfg文件，权重保存为.weights文件。我使用OpenCV，它是在Raspberry Pi 3b+上与Openvino一起安装的。我的代码如下：

import cv2
# load network
ocr_net  = cv2.dnn.readNetFromDarknet('ocr-net.cfg', 'ocr-net.weights')
ocr_net.setPreferableTarget(cv2.dnn.DNN_TARGET_MYRIAD)
ocr_net.setPreferableBackend(cv2.dnn.DNN_BACKEND_INFERENCE_ENGINE)
# perform inference
im = cv2.imread('img_path.jpg')
blob = cv2.dnn.blobFromImage(im, swapRB=False)
ocr_net.setInput(blob)
result = ocr_net.forward()

在运行此代码时，我收到以下错误：

Traceback (most recent call last):
File "test.py", line 12, in <module>
result = ocr_net.forward()
cv2.error: OpenCV(4.1.2-openvino) /home/jenkins/workspace/OpenCV/OpenVINO/build/opencv/modules/dnn/src/op_inf_engine.cpp:704: error: (-215:Assertion failed) Failed to initialize Inference Engine backend: Device with "CPU" name is not registered in the InferenceEngine in function 'initPlugin'

如果我将推理目标更改为cv2.dnn.DNN_TARGET_CPU，将推理后端更改为cv2.dnn.DNN_BACKEND_OPENCV，则模型工作正常，并且能够正确执行 OCR。我的openvino/openCV设置也正确安装，因为我可以使用openCV(使用readNetFromTensorflow()/readNetFromCaffe()(正确运行其他Tensorflow/Caffe模型。

我尝试使用的OCR网络可以在这里找到： cfg ， 权重 .

任何帮助不胜感激！

Answer 1

根据我对英特尔文档的记忆，opencv.dnn 的 openVINO 发行版无法处理暗网模型。在此页面上，您可以找到我认为可以解释为所有支持的框架的列表。

我目前正在尝试将我的暗网自定义模型转换为 tf.pb 表示形式，到目前为止没有运气。