如何在C++中获取caffe的层类型

3

是否可能在C++中获取每个层的类型(例如:卷积、全连接、数据等)和顶部标签(例如:ip1、ip2、conv1、conv2)?

我搜索了提供的示例,但没有找到任何内容。目前我只能通过以下命令获得层名称:

cout << "Layer name:" << "'" << net_->layer_names()[layer_index]

我正在寻找类似于 net_->layer_type 的命令。谢谢!
1个回答

1

Layer类有一个公共成员函数virtual const char *type(),返回层类型。因此

cout << "Layer type: " << net_->layers()[layer_index]->type();

应该可以解决问题。

它奏效了。我知道你提到的概念,但无法实现(调用函数)。在哪里可以找到关于如何调用所有函数或使用C++中"caffe"框架的教程? 感谢您的帮助。 - Ganesh M S
1
@GaneshMS 我不太确定你的意思,但是你可以在这里找到Caffe中的类列表,例如:http://caffe.berkeleyvision.org/doxygen/annotated.html。层类型自然由“Layer”类知道,因此检查其公共成员函数,您可以找到所需类型的函数。同样,从“class Net”中,您可以找到获取层向量的公共函数。 - Ari Hietanen

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接