将.h5模型转换为TFlite,找不到文件路径

时间:2020-10-03 19:37:20

标签: spyder tensorflow2.0 tensorflow-lite h5py kaggle

我们正在研究一种机器学习keras张量流模型。我们希望将我们保存的.h5文件转换为tflite文件,以最小化模型的大小。这是我们在spyder上将.h5转换为tflite的代码。

import tensorflow as tf

new_model = tf.keras.models.load_model('C:/Users/Edward Z/Downloads/Google-Image-Scraper-master/LymeModel.h5')
converter = tf.lite.TFLiteConverter.from_keras_model(new_model)
tflite_model = converter.convert()
# Save the model.
with open('Lymemodel.tflite', 'wb') as f:
  f.write(tflite_model)

输出为

INFO:tensorflow:Assets written to: C:\Users\EDWARD~1\AppData\Local\Temp\tmp9ieudidr\assets

我尝试为用户寻找文件路径,但是找不到文件。我们还尝试在kaggle笔记本电脑上转换h5文件,但该文件太大,因此我们转到了spyder。如果有人可以为此提供帮助,那就太好了!

0 个答案:

没有答案