tensorflow Cpp获取网络每一层的名字

在之前的博客中提到了使用tensorflow的C++接口调用已经训练好的模型pb文件，这一篇简单说下获取图模型每一层的名称，虽然不一定有用。

前面就知道在tensorflow的C++接口中，用GraphDef来定义一个图模型的类，然后通过ReadBinaryProto()函数来从pb文件中加载图模型结构，这个时候，模型的每一层在GraphDef中就是一个结点(node)，所以有多少层(结点)可以用成员函数node_size()来获取，然后第i层(结点)通过成员函数node(int)来获取，再然后名字当然是成员函数name()啦，name()返回的是一个const std::string的字符串，通过它可以知道层的名字，有什么用呢，可以通过名字来获取层的输出的张量(前面就是通过名字来获取模型的输入和输出层，进而获取模型对输入张量的的输出)。

代码如下：

Status status_load = ReadBinaryProto(Env::Default(), model_path, &graphdef);
if (!status_load.ok())
{
    cout << "ERROR: Loading model failed..." << model_path << std::endl;
    cout << status_load.ToString() << "\n";
    system("pause");
    return -1;
}
for (int i=0; i < graphdef.node_size(); i++)
{
    std::string name = graphdef.node(i).name();
    std::cout << name << std::endl;
}

好了，这样就可以输出加载的模型的每一层的名字。

 

月亮陪着我走回家。
我想把她带到将来的日子里去。
一路静悄悄……

 -- 顾城