tensorflow pb模型转tflite模型

说明：2018年8月8号，tensorflow-gpu版本是1.9.0，平台是ubuntu16.04。

目前windows平台lite文件夹内文件不全，经过努力，还是失败，转战ubuntu发现可行，暂且可以确定ubuntu16.04可以用代码把pb文件转到tflite文件。代码如下

import tensorflow as tf
path="ckpt_to_pb_model.pb"        #pb文件位置和文件名
inputs=["inputs"]               #模型文件的输入节点名称
classes=["classes"]            #模型文件的输出节点名称
converter = tf.contrib.lite.TocoConverter.from_frozen_graph(path, inputs, classes)
tflite_model=converter.convert()
open("/home/zhang/anaconda3/model_pb.tflite", "wb").write(tflite_model)

tensorflow1.8.0以前版本不支持pb转tflite。

ckpt转pb文件的博客以后更新！