[英]How to convert a .h5 model to a .tflite and also get image labels?
我對深度學習很陌生,我有一個非常大的自定義數據集,如何將.h5 model 轉換為 a.tflite model 以及如何在不手動操作的情況下生成所有標簽?
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir) # path to the SavedModel directory
tflite_model = converter.convert()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.