简体   繁体   中英

How to convert a .h5 model to a .tflite and also get image labels?

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?

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()

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM