如何将 .h5 模型转换为 .tflite 并获取图像标签?

时间:2021-02-03 13:32:36

标签: python tensorflow keras deep-learning

我是深度学习的新手,我有一个非常大的自定义数据集,如何将 .h5 模型转换为 .tflite 模型,以及如何在不手动生成所有标签的情况下生成所有标签?

>

1 个答案:

答案 0 :(得分:0)

From Tensorflow documentation

转换模型

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir) # path to the SavedModel directory
tflite_model = converter.convert()
相关问题