将Tensorflow模型导出到Android

Question

我最初有一个用于生成和训练CNN模型的代码，没有用于保存或编写图形的附加代码。

但是，我在Android Studio中不接受使用此代码生成的优化.pb文件，并且仿真停止。

我所做的是以下内容：

1. 使用我的基本代码生成CNN模型，我在训练部分添加了这些行：



1. 我通过添加＆＃34; name＆＃34;来编辑输入占位符和输出预测。属性，

X = tf.placeholder(tf.float32, shape [None,input_height,input_width,num_channels], name="input")

y_ = tf.nn.softmax(tf.matmul(f, out_weights) + out_biases, name='y_')

因为我认为创建图表需要这些。除了这个

之外，我没有将任何其他代码添加到我的原始代码中
1. 然后在保存模型后，生成.pbtxt和.ckpt文件后，冻结模型并生成.pb文件



我把这段代码放在

之外

in with session as tf.Session():

训练结束后。

1. 然后你仍然需要优化这个.pb文件



我把这段代码放在

之外

in with session as tf.Session():

训练结束后。

将放置在Android Studio项目的资源文件夹中的最终优化模型为optimized_frozen_har.pb

但仿效停止了。我正在使用手机。如果我使用原始optimized_frozen_har.pb文件进行模拟，则不会出现任何错误。原始optimized_frozen_har.pb文件的文件大小为21kB，但我生成的文件大小仅为4kB。

可能存在生成冻结和优化模型的代码问题。除了添加＆＃34;名称＆＃34;输入和输出的属性，我应该在原始代码中添加其他代码行吗？ （例如with namescope..等