我正在尝试使用c++在caffe框架中读取重量和偏见。这是我的代码

shared_ptr<Blob<float> >& weight = current_layer->blobs()[0];//for weights
shared_ptr<Blob<float> >& bias = current_layer->blobs()[1];//for bias

但是,如果对于某些模型,不存在或没有定义偏差,则通过细分错误来确定偏差。

那么哪个函数返回一个 bool(boolean) 值,该值指示存在偏见以及如何在c++中调用该函数?

最佳答案

blobs返回的current_layer->blobs()存储在 std::vector 中,您可以使用其size属性:

if (current_layer->blobs().size() > 1) {
    shared_ptr<Blob<float> >& bias = current_layer->blobs()[1];//for bias
}

有关更多详细信息,请参见this similar answer以获取python接口(interface)。

07-26 09:36