如何用C ++而不是prototxt初始化卷积层？

Question

我试图将caffe提供的图层类“CuDNNConvolutionLayer”导入到我自己的项目（纯C ++项目）中。因此我无法使用原型文本来指定图层。相反，我必须在声明时手动将参数发送到图层。 （我不确定这种方式是否正确，我只是在探索）

我这样做了：

int param={20,5,1,0,1,1,???,???}; caffe::CuDNNConvolutionLayer<float> cv1(*param);

param中的最后两个参数设置为初始化“bias_term”（true或false）和“engine”（caffe或CUDNN）。正如我所提到的，我不能忽略这两个值，因为这不是原型，所以如果我想使用CUDNN引擎并且没有偏差项，我应该在这里填充什么值？ （我已经将param声明为int，所以我认为并且期望答案应该是0，1-希望如此）

Operating system: ubuntu 18.04
Compiler: gcc
CUDA version (if applicable): 10.0
CUDNN version (if applicable): v7
BLAS:
Python version (if using pycaffe):
MATLAB version (if using matcaffe):

Answer 1

CuDNNConvolutionLayer构造函数接受对LayerParameter protobuf对象（不是int *）的引用。因此，您必须使用默认构造函数创建一个空的LayerParameter对象，并使用其setter方法向其添加必要的字段。据我所知，ProtoBuf不支持{} -style初始化。

初始化LayerParameter的另一种方法是使用ProtoText表示解析字符串文字。

阅读"ProtoBuf C++ generated code"的文档将有所帮助。

另外，请查看source of Scale layer以获取FillerParameter和LayerParameter的创建示例以构建图层。