将ckpt文件生成pb文件的详细过程,并调用pb文件进行模型预测.

前段时间搭建了一个分类网络模型,而后用本身的数据进行了800epoch’的训练,最后默认生成了三个ckpt文件.因为要同时运行几个网络,因此打算将这个网络模型进行固化成pb文件,而后直接调用. 主要包括一下内容:前端 1.查看ckpt模型的输入输出张量名称. 2.将ckpt文件生成pb文件. 3.查看生成的pb文件的输入输出节点 4.运行pb文件,进行网络预测 1.查看ckpt网络的输入输出张量名

>>阅读原文<<