由.pd生成.tflite

最新推荐文章于 2021-12-29 16:05:24 发布

我叫夏满满

最新推荐文章于 2021-12-29 16:05:24 发布

阅读量336

点赞数

文章标签： tensorflow python

本文链接：https://blog.csdn.net/qq_42479394/article/details/112916565

版权

这篇博客介绍了如何使用TensorFlow的TFLiteConverter将已保存的模型转换为TFLite格式，便于在移动端和嵌入式设备上部署。通过指定模型的保存路径，调用TFLiteConverter，进行转换，并将生成的TFLite模型保存到桌面。

摘要由CSDN通过智能技术生成

比如我的模型在"/tmp/test_saved_model"，最后生成文件就在桌面上

import tensorflow as tf
#把pb文件路径改成自己的pb文件路径即可
saved_model_dir="/tmp/test_saved_model"
# Convert the model
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir) # path to the SavedModel directory
tflite_model = converter.convert()

# Save the model.
with open('model.tflite', 'wb') as f:
  f.write(tflite_model)