将pb文件转化为tflite文件

最新推荐文章于 2025-03-24 23:30:00 发布

HK_Joe

最新推荐文章于 2025-03-24 23:30:00 发布

阅读量6.5k

点赞数 6

分类专栏： Python 文章标签： tflite 迁移学习 mobileNet 迁移到安卓端 Android

本文链接：https://blog.youkuaiyun.com/HK_Joe/article/details/102730752

版权

最近在做一个关于MobileNet的卡片识别任务，最后的目的在于把训练好的模型移植到手机端，关于的移植的步骤，网上已经讲了很多了，我就不多讲述了，这里推荐一个博客我的模型移植就是参考这里的，同时我的模型是直接通过TFhub上的MobileNet迁移学习得到的。迁移之后会生成一个pb文件，然而pb文件又不能直接使用Tensorflow-lite这一个神器，因此就必须把pb模型转化为tflite模型，在转化的时候猜到了一个坑，苦苦搜寻终于在StackOverflow上找到了解决办法，附上链接。也贴上我错误的代码：

import tensorflow as tf

in_path=r'D:\tmp_mobilenet_v1_100_224_classification_3\output_graph.pb'
out_path=r'D:\tmp_mobilenet_v1_100_224_classification_3\output_graph.tflite'

input_tensor_name='Placeholder'
input_tensor_shape={'Placeholder':[1,224,224,3]}

class_tensor_name='final_result'


convertr=tf.lite.TFLiteConverter.from_frozen_graph(in_path,input_arrays=input_tensor_name