我正在尝试生成要用于移动应用程序的自定义张量流模型(tf / tflite文件)。
我浏览了一些机器学习和张量流博客,从那里开始生成一个简单的ML模型。
所有这些都非常好,它们指导我执行以下步骤,
i)安装所有必需的工具(TensorFlow,Python,Jupyter等)。 ii)加载培训和测试数据。 iii)运行张量流训练,并评估结果。 iv)提高准确性的步骤
但是我无法生成.tf / .tflite文件。
我尝试了以下代码,但是生成了一个空文件。
converter = tf.contrib.lite.TFLiteConverter.from_session(sess,[],[])
model = converter.convert()
file = open( 'model.tflite' , 'wb' )
file.write( model )
我检查了stackoverflow中的几个答案,根据我的理解,为了生成.tf文件,我们需要创建pb文件,冻结pb文件然后生成.tf文件。
但是我们如何实现呢?