c++ - 我怎么知道使用caffe framwork和c++程序的层中是否不存在偏差

Question

我正在尝试使用 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 值来指示偏差的存在以及如何在 c++ 中调用该函数？

Answer 1

从 current_layer->blobs() 返回的 blob 存储在 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)以获取更多详细信息。