tensorflow深度学习实战笔记（二）：把训练好的模型进行固化

最新推荐文章于 2025-07-07 10:20:13 发布

原创

最新推荐文章于 2025-07-07 10:20:13 发布 · 7.1k 阅读

23 ·

CC 4.0 BY-SA版权

一、导出前向传播图

二、对模型进行固化

三、pb文件转tflite文件

说明：接我的上一篇博客：tensorflow深度学习实战笔记（一）：使用tensorflow slim自带的模型训练自己的数据

前面讲解了如何用tensorflow slim训练自己的模型，现在讲解如何把训练好的模型的cpkt文件固化成pb文件（即最终的模型）

一、导出前向传播图

在slim文件夹下有export_inference_graph.py文件（slim文件夹的位置可以参考我的上一篇博客），运行该脚本即可导出前向传播图，运行方式如下：

python export_inference_graph.py \
  --alsologtostderr \
  --dataset_dir=/home/yuping-chen/slim/my_data/fruits/ \#数据集的路径
  --dataset_name=fruit \#数据集的名字
  --model_name=inception_v3 \#导出的模型
  --image_size=224 \#图片尺寸
  --output_file=my_model/inception_v3_inf.pb#输出文件名，可以自定义