是否可能在C++中获取每个层的类型(例如:卷积、全连接、数据等)和顶部标签(例如:ip1、ip2、conv1、conv2)?
我搜索了提供的示例,但没有找到任何内容。目前我只能通过以下命令获得层名称:
cout << "Layer name:" << "'" << net_->layer_names()[layer_index]
我正在寻找类似于 net_->layer_type 的命令。谢谢!
是否可能在C++中获取每个层的类型(例如:卷积、全连接、数据等)和顶部标签(例如:ip1、ip2、conv1、conv2)?
我搜索了提供的示例,但没有找到任何内容。目前我只能通过以下命令获得层名称:
cout << "Layer name:" << "'" << net_->layer_names()[layer_index]
Layer
类有一个公共成员函数virtual const char *type()
,返回层类型。因此
cout << "Layer type: " << net_->layers()[layer_index]->type();