我尝试在https://codelabs.developers.google.com/codelabs/tensorflow-for-poets-2的示例中替换graph.pb文件 但它未能在Andriod中发布错误:
不是有效的TensorFlow图形序列化:NodeDef提到了attr' dilations'不在Op名称= Conv2D。
12-16 15:06:24.986 4310-4310/org.tensorflow.demo E/AndroidRuntime: Caused by: java.io.IOException: Not a valid TensorFlow Graph serialization: NodeDef mentions attr 'dilations' not in Op<name=Conv2D; signature=input:T, filter:T -> output:T; attr=T:type,allowed=[DT_HALF, DT_FLOAT]; attr=strides:list(int); attr=use_cudnn_on_gpu:bool,default=true; attr=padding:string,allowed=["SAME", "VALID"]; attr=data_format:string,default="NHWC",allowed=["NHWC", "NCHW"]>; NodeDef: conv0/Conv2D = Conv2D[T=DT_FLOAT, data_format="NHWC", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 2, 2, 1], use_cudnn_on_gpu=true](truediv, conv0/W). (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).
at org.tensorflow.contrib.android.TensorFlowInferenceInterface.loadGraph(TensorFlowInferenceInterface.java:392)
如何使用正确的Conv2D graphDef生成推理pb文件?
答案 0 :(得分:5)
我遇到了同样的问题并试图遵循上述解决方法。 (没有成功)
但后来我重新评估了原来的错误。 &#34;版本未与Tensorflow匹配&#34;
这引导我(显而易见的)简单的解决方案对我有用。
验证笔记本电脑上使用的Tensorflow版本是否构建1.5.0
在Android版本中设置相同版本。拍前额....
dependencies {
compile 'org.tensorflow:tensorflow-android:1.5.0'
}
答案 1 :(得分:1)
我有同样的问题。正如here所述,此错误似乎是由于用于运行训练脚本的TensorFlow版本与Android应用程序中包含的库版本之间的版本不匹配所致。
要解决此问题,您可以尝试以下操作:
libandroid_tensorflow_inference_java.jar
in libtensorflow_demo.so
和libtensorflow_inference.so
文件(对应于该架构)在build.gradle中,设置nativeBuildSystem = 'none'
并修改最后的“依赖项”,如下所示:
依赖{ if(nativeBuildSystem =='cmake'|| nativeBuildSystem =='none'){ 实现文件('libs / libandroid_tensorflow_inference_java.jar') } }
编译,上传到测试设备并运行应用程序。它对我有用。