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环境
详情见快速创建离线推理Ai1S环境 - Wiki - Gitee.com
ATC模型转换pb为om
详情见