Как создать Tenorflow-Lite модель.bin для Android NN API

Question

Как создать модель tenorflow 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)