我正在使用Tensorflow face detection model from OpenCV's DNN module至cv2.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中运行它,对吧?但是如何?