如何为android NN api创建tensorflow-lite模型.bin

Question

如何创建tensorflow lite模型

在下面的链接中提供？

https://github.com/googlesamples/android-ndk/tree/master/nn_sample/app/src/main/assets

Answer 1

您可以按照https://www.tensorflow.org/mobile/tflite/devguide

中的文档创建模型。

对于NN-API示例，您可以执行以下操作：

import tensorflow as tf
t= tf.placeholder(name="t", dtype=tf.float32, shape=(2,1))
w=tf.constant([3.0,4.0])

temp=t+w
val=temp[0]*temp[1]

out = tf.identity(val, name="out")

with tf.Session() as sess:
   tflite_model = tf.contrib.lite.toco_convert(sess.graph_def, [t], [out])
   open("converteds_model.tflite", "wb").write(tflite_model)