.pb转化成.tflite

    xiaoxiao2024-11-22  74

    1.创建训练图,其中加入create_training_graph(),训练。把训练好的数据存在checkpoint file中。 2. 创建推理图,其中加入create_eval_graph()。把训练好的数据导入推理图中,即load checkpoint。 3. Freeze 生成.pb file.(演讲的例子把2,3合成了一个文件中做。) 4. 在linux machine上把这个.pb file转化成.tflite file。 5. 可以用tflite python api的 interpreter读入这个.tflite模型然后做testing. 6. https://www.tensorflowers.cn/t/7136

    最新回复(0)