如何在Tensorflow中使用OpenCV DNN人脸检测器模型?

时间:2018-11-10 09:14:38

标签: opencv tensorflow face-detection

我正在使用Tensorflow face detection model from OpenCV's DNN modulecv2.dnn.readNetFromTensorflow()。现在,我想直接在Tensorflow中进行推断,以使其在GPU上运行。我可以在Tensorflow中加载模型,但无法弄清楚如何将图像输入模型并检索结果。

第一个问题:.pbtxt文件和.pb文件中的节点似乎有所不同。在.pbtxt文件中,我可以看到应为结果的“ detection_out”节点。 .pb文件中找不到该节点。

KeyError: "The name 'detection_out:0' refers to a Tensor which does not exist. The operation, 'detection_out', does not exist in the graph."

那我该如何检索结果?

第二个问题:当我将输出更改为.pb文件中存在的节点时,我收到另一条错误消息:

InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'Dequantize' with these attrs.  Registered devices: [CPU,GPU], Registered kernels:

该模型在OpenCV中运行良好。由于它是Tensorflow模型,因此应该可以直接在TF中运行它,对吧?但是如何?

0 个答案:

没有答案