1

我们正在尝试将图像输入到我们的 tensorflow 模型中，但我们不断收到此错误：

错误第 1 部分

错误第 2 部分

我们最初使用 pytorch，将其转换为 ONNX 模型，然后转换为用于 Android Studio 的 tensorflow 模型。我们能够成功加载模型，但之后我们遇到了错误。

提前感谢那些回复的人！

4

2 回答 2

0

@AEye Alliance 我在使用旧版本的 tensor flow lib for android 时遇到了同样的问题，即 compile 'org.tensorflow:tensorflow-android:1.5.0' 我将其替换为 compile 'org.tensorflow:tensorflow-android:1.10。 0' 因为 'GatherV2' 在更新的库中可用。请让我知道它是否解决了您的问题。

于 2018-09-08T12:51:12.950 回答
0

您需要使用 --copt=-D__ANDROID_TYPES_FULL__ 重建 tensorflow_inference_android 库。这将使它能够在 int64 操作上工作。但它会慢得多。

于 2018-11-17T06:08:43.360 回答