gpt4 book ai didi

c++ - 如何在 C++ 中获取层的顶部标签?

转载 作者:行者123 更新时间:2023-11-30 05:08:11 24 4
gpt4 key购买 nike

1)是否有可能在 C++ 中获取每一层的顶部标签(例如:ip1、ip2、conv1、conv2)?如果我的图层是

layer {
name: "relu1_1"
type: "Input"
top: "pool1"
input_param {
shape: {
dim:1
dim: 1
dim: 28
dim: 28
}
}
}

我想获得顶部标签,在我的例子中是“pool1

我搜索了提供的示例,但找不到任何内容。目前我只能通过以下命令获取图层名称和图层类型,

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

2) 我在哪里可以找到教程或示例来解释使用 c++ 使用 caffe 框架的最常用 API?

提前谢谢你。

最佳答案

Net doxygen 中的类:

const vector< vector< Blob< Dtype > * > > all_ tops = net_->top_vecs();  // get "top" of all layers
Blob<Dtype>* ptop = all_tops[layer_index][0]; // pointer to top blob of layer

如果你想要图层的名字,你可以

const string layer_name = net_->layer_names()[layer_index];

您可以使用net_ 接口(interface)访问各种名称/数据,只需阅读doc !

关于c++ - 如何在 C++ 中获取层的顶部标签?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46970653/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com