在android中找不到TF模型输入节点

Question

我训练了一个头发识别模型（.pb文件）。它可以很好地使用python代码。输入节点是“hair_recognition/input_1”，但是当我在android中使用该模型时，我得到以下错误：

java.lang.IllegalArgumentException: No Operation named [hair_recognition/input_1]

我已经通过在Python和Android中打印节点名来调试模型。但无法找出原因。 有什么建议可以引起这种行为吗？

代码加载模型和获取输入张量节点

Python:

^{pr2}$

安卓系统：

inferenceInterface.feed(
                INPUT_NODE, floatValues, 1, bitmap.getWidth(), bitmap.getHeight(), 3);

Answer 1

我之前也遇到过类似的问题。 原因是在python中加载带有前缀string-hair_recognition的模型。所以你的输入张量就是这个名字。 将前缀更改为其他值，如“.”或空字符串。在这种情况下，它也应该起作用。在

在Android中，将input_1:0而不是hair_recognition/input_1:0传递给INPUT_NODE。错误应该得到解决。在