Jupyter Notebook中安装tflite-model-maker失败求助

一、Android Studio提示模型非TFLite格式的修复 #

• 确认转换流程合规：必须通过TensorFlow Lite Converter生成模型，不能直接重命名文件。转换示例代码：

  import tensorflow as tf
  
  # 加载训练好的Keras模型
  model = tf.keras.models.load_model("your_trained_model.h5")
  # 转换为TFLite格式
  converter = tf.lite.TFLiteConverter.from_keras_model(model)
  tflite_model = converter.convert()
  # 保存为.tflite文件
  with open("text_classifier.tflite", "wb") as f:
      f.write(tflite_model)
  

• 验证文件有效性：安装tflite-support后使用tflite-cli工具检查：

  tflite_analyze --model_path=text_classifier.tflite
  

  输出模型结构信息则文件有效，报错则重新执行转换流程。
• Android项目配置：将.tflite文件放在src/main/assets目录，在build.gradle中添加避免压缩的配置：

  android {
      aaptOptions {
          noCompress "tflite"
      }
  }
  

二、Google Colab安装tflite-model-maker卡住的解决 #

• 先重启Colab运行时：点击菜单栏「Runtime」→「Restart runtime」清空缓存。
• 指定版本安装，规避最新版依赖冲突：

  !pip install tflite-model-maker==0.4.2
  

• 若仍卡住，先升级pip再安装：

  !pip install --upgrade pip
  !pip install tflite-model-maker==0.4.2
  

三、Jupyter Notebook安装依赖的版本兼容修复 #

• 匹配Python版本：tflite-model-maker 0.4.2要求Python 3.7-3.9，更高版本会出现依赖缺失问题。用conda创建虚拟环境：

  conda create -n tflite_env python=3.9
  conda activate tflite_env
  

  激活环境后安装Jupyter：

  pip install jupyter
  

• 安装兼容版本依赖：直接指定版本安装，避免自动拉取不兼容包：

  %pip install tflite-model-maker==0.4.2 tflite-support==0.4.2
  

• 解决subprocess错误：安装预编译包跳过源码编译：

  pip install --only-binary :all: tflite-model-maker==0.4.2
  

新手快速构建文本分类TFLite模型步骤 #

1. 在Colab或虚拟环境中运行以下代码：

   from tflite_model_maker import text_classifier
   from tflite_model_maker.text_classifier import DataLoader
   
   # 加载CSV格式数据集（每行格式：标签\t文本）
   data = DataLoader.from_csv(
       filename="your_dataset.csv",
       text_column="text",
       label_column="label"
   )
   # 拆分训练/测试集
   train_data, test_data = data.split(0.8)
   # 训练模型
   model = text_classifier.create(train_data)
   # 评估模型性能
   loss, accuracy = model.evaluate(test_data)
   # 导出TFLite模型
   model.export(export_dir=".", export_format="tflite")
   

2. 将导出的tflite文件导入Android项目assets目录，按官方文档集成推理代码即可。

内容的提问来源于stack exchange，提问作者Skyain