TF.KERAS HDF5模型和Keras HDF5模型



我想将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)

tf.keras hdf5模型和keras hdf5型号并非不同,除了不可避免的软件版本更新同步性。这就是官方文档所说的:

tf.keras是TensorFlow的Keras API规范的实现。这是一个高级API,用于构建和训练模型,其中包括对特定功能的一流支持

如果转换器可以将KERAS模型转换为tf.lite,则将提供相同的结果。但是TF.Lite功能比TF.Keras更有限。如果此功能集对您来说不足,那么您仍然可以使用TensorFlow,并享受其其他优势。

可能是,您的模型可以在智能手机上运行不久。

最新更新