2025-11-20

训练模型

目标： 训练一个模型，能够识别图片中人成年还是儿童 可导出给安卓ios客户端使用的模型文件.tflite文件 也可到导出给web浏览器使用的.json和.bin文件

数据集

线上数据

模型

使用tensorflow框架，使用mobilenet_v2模型，使用adam优化器，使用交叉熵损失函数，使用准确率作为评估指标。

训练

使用线上数据训练模型，使用线上数据验证模型。

评估

使用线上数据评估模型。

1. 使用python3.9环境，下载 https://www.tensorflow.org/install/pip#macos 安装tensorflow
2. 执行训练脚本 python train.py
3. 导出模型文件 model.export("saved_model")
4. saved_model文件夹中包含的就是核心格式，都是基于这个文件进行转换的

转换为 Web 端（TensorFlow.js） 进入到saved_model文件夹，执行以下命令，将模型文件转换为浏览器使用

tensorflowjs_converter \
  --input_format=tf_saved_model \
  --output_format=tfjs_graph_model \
  --signature_name=serving_default \
  saved_model \
  web_model

转换完成后 web_model/ 就是你浏览器要加载的文件夹。

转换为 Android 端（TensorFlow Lite） 进入到saved_model文件夹，执行以下命令，将模型文件转换为安卓使用

tflite_convert \
  --saved_model_dir=saved_model \
  --output_file=model.tflite

转换完成后 model.tflite 就是你安卓要加载的文件。