NPU离线推理指南

以下仅为个人经验，不懂及时百度。

转化ckpt文件为pb文件

关键：input和output要取名，函数名称freeze_graph。
tf.train.write_graph(sess.graph_def, ‘./pb_model’, ‘model.pb’)
freeze_graph.freeze_graph(input_graph=’./pb_model/model.pb’,
output_node_names=‘output’,
output_graph=’./pb_model/deeplabv2.pb’,)
其中三个为必填项，input_graph为与write_graph保持一致，output_node_names为输出名称，output_graph为pb文件存放地点。

检验pb文件

略，就是模型加载pb代替加载ckpt代码。

创建310环境

详情见