Я использую модель распознавания лиц Tensorflow из модуля DNN OpenCV - cv2.dnn.readNetFromTensorflow(). Теперь я хотел бы сделать вывод непосредственно в Tensorflow, чтобы он работал на GPU. Я могу загрузить модель в Tensorflow, но не могу понять, как передать изображение в модель и получить результаты.
Первая проблема: узлы в файле .pbtxt и в файле .pb кажутся разными. В файле .pbtxt я могу видеть узел 'creation_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, верно? Но как?