| 123456789101112 |
- import tensorflow as tf
- # 加载已保存的 TensorFlow 模型（saved_model.pb 文件所在的文件夹路径）
- saved_model_dir = "D:\STUDY\Project\langchain\jizhouyao\mobilenet_model.pb" # 这是你上传的模型目录路径
- converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
- # 可选：优化模型大小和性能
- converter.optimizations = [tf.lite.Optimize.DEFAULT]
- converter.target_spec.supported_types = [tf.float16]
- # 转换为 TensorFlow Lite 格式
- tflite_model = converter.convert()
- # 保存为 .tflite 文件
- with open("mobilenet_model.tflite", "wb") as f:
- f.write(tflite_model)
|
