Question

从documentation我认为有一个构造函数采用NetParameter参数，

显式网络（const NetParameter＆amp; param）;

但是当我尝试使用它时：

    import caffe
    from caffe import layers as L
    from google.protobuf import text_format

def logreg(hdf5, batch_size):
    # logistic regression: data, matrix multiplication, and 2-class softmax loss
    n = caffe.NetSpec()
    n.data, n.label = L.HDF5Data(batch_size=batch_size, source=hdf5, ntop=2)
    n.ip1 = L.InnerProduct(n.data, num_output=2, weight_filler=dict(type='xavier'))
    n.accuracy = L.Accuracy(n.ip1, n.label)
    n.loss = L.SoftmaxWithLoss(n.ip1, n.label)
    return n.to_proto()

logreg_str = str(logreg('examples/hdf5_classification/data/test.txt', 10))

net_param = caffe.proto.caffe_pb2.NetParameter()
_ = text_format.Merge(logreg_str, net_param)

print type(net_param);
caffe.Net(net_param, caffe.TEST)

ipython中出现以下错误

<class 'caffe.proto.caffe_pb2.NetParameter'>

--------------------------------------------------------------------------- 
ArgumentError                         Traceback (most recent call last)
<ipython-input-20-edce76ff13a1> in <module>()
     14 
     15 print type(net_param);
---> 16 caffe.Net(net_param, caffe.TEST)

ArgumentError: Python argument types in
    Net.__init__(Net, NetParameter, int) did not match C++ signature:
    __init__(boost::python::api::object, std::string, std::string, int)
    __init__(boost::python::api::object, std::string, int)

那我在这里做错了什么？我如何使用此构造函数？

注意：我知道如何使用来自磁盘构造函数的＆＃34;读取文件＆＃34;已经，我想使用NetParameter /或理解为什么它不起作用。

在Shai的评论后编辑：

我在2015年7月26日使用此命令获得了caffe： git clone https://github.com/BVLC/caffe.git

这是我磁盘上的文件：

~/caffe/src/caffe$ grep NetParameter net.cpp | head -1
Net<Dtype>::Net(const NetParameter& param) {
~/caffe/src/caffe$ ~/caffe/build/tools/caffe -version
caffe

-version开关似乎什么都不做。我浏览了源代码，无法找到版本号。

Answer 1

您的代码没有错。实际上，在C ++中有一个Net类的重载构造函数，但它目前没有被python接口公开。 python接口仅限于带有文件参数的构造函数。

我不确定是否只是在python / caffe / _caffe.cpp中公开它是唯一阻止我们用NetParameter构建python Net对象的东西，或者是否需要更精细的更改。

Answer 2

我遇到了同样的问题，我得到了一个解决方案on google user group，这解释了你的c ++ boost lib太旧了，你可能需要更新它。

使用NetParameter构造caffe.Net对象

2 个答案: