我们正在尝试将图像输入到我们的张量流模型中,但我们不断收到此错误:
我们最初使用pytorch,将其转换为ONNX模型,然后转换为可在Android Studio中使用的tensorflow模型。我们能够成功加载模型,但是之后我们会遇到错误。
预先感谢做出回应的人!
答案 0 :(得分:0)
@AEye Alliance我在使用较旧版本的tensor flow lib for android时遇到了相同的问题,即编译'org.tensorflow:tensorflow-android:1.5.0'我将其替换为编译'org.tensorflow:tensorflow-android :1.10.0”,因为更新的库中提供了“ GatherV2”。请让我知道它是否解决了您的问题。
答案 1 :(得分:0)
您需要使用以下命令重建tensorflow_inference_android库 --copt = -D__ANDROID_TYPES_FULL__。这将使其能够在int64 ops上运行。但这会慢很多。