如何将 a.h5 model 转换为 a.tflite 并获取图像标签？

Question

I am pretty new to Deep Learning, I have a custom dataset which is quite large, how do I convert the.h5 model to a.tflite model and also how do I generate all the labels without doing it manually?

Answer 1

From Tensorflow documentation

Convert the model

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