Android中运行Tensorflow程序3-遇到的错误及解决

最新推荐文章于 2023-04-12 09:16:16 发布

原创

最新推荐文章于 2023-04-12 09:16:16 发布 · 3.6k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#android #Tensorflow #深度学习

在Android中运行TensorFlow程序时遇到了两个错误：A/native: tensor.cc:487 Check failed错误和java.lang.UnsatisfiedLinkError。前者由于模型中数据类型不匹配引起，解决方案是将模型中的数据类型调整为匹配。后者由于so文件路径配置不当导致，解决办法是将所有TensorFlow库的so文件放置在正确的位置，并修正配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

A/native: tensor.cc:487 Check failed: dtype() == expected_dtype (9 vs. 3)

原因解释

出现这种错误的原因是：程序预期的数据是tf.int32，然而真正的数据是tf.int64。在我的应用中，出现以上错误的程序如下。
在TF模型中，有如下的定义。

self.predictions = tf.argmax(self.logits, 1, name="predictions")

这里，函数tf.argmax返回的Tensor的类型是tf.int64。
在进行了freeze_graph和optimize_for_inference之后，我在Android app中按照如下方式调用模型进行了预测：

    tensorFlowInferenceInterface = new TensorFlowInferenceInterface();
    tensorFlowInferenceInterface.initializeTensorFlow(assetManager,MODEL_FILE);
    tensorFlowInferenceInterface.fillNodeInt(INPUT_X,INPUT_X_SIZE,input);
    tens