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)