我想将Keras模型转换为Tensorflow Lite模型。当我查看文档时,有人说我们可以使用tf.keras HDF5模型作为输入。这是否意味着我可以使用保存的HDF5 Keras模型作为它的输入,或者tf.keras HDF5模型和Keras HDF5模型是不同的东西?
文档:https://www.tensorflow.org/lite/convert
编辑:我可以使用这个API将我的Keras模型转换成Tensorflow Lite模型,但是我还没有测试它。我的代码:
converter = tf.lite.TFLiteConverter.from_keras_model_file(path + 'plant-
recognition-model.h5')
tflite_model = converter.convert()
with open('plant-recognition-model.tflite', 'wb') as f:
f.write(tflite_model)发布于 2019-05-04 16:12:39
tf.keras HDF5模型和Keras HDF5模型除了不可避免的软件版本更新的同步性外,没有什么不同。官方文件是这么说的
tf.keras是TensorFlow对Keras规范的实现。这是一个用于构建和训练模型的高级API,包括对TensorFlow特定功能的一流支持。
如果转换器能够将keras模型转换为tf.lite,它将提供相同的结果。但是tf.lite的功能比tf.keras更有限。如果这个特性集对您来说还不够,您仍然可以使用tensorflow,并享受它的其他优势。
也许,你的模型很快就能在智能手机上运行。
https://stackoverflow.com/questions/55964427
复制相似问题