caffe 之 proto 理解

caffe 所有层的参数全部定义在一个caffe.proto 的文件里面,里面有每个参数的详细解释,点击打开链接

在protocol buffers的官网的教程中,看到了一些基本的语法和编译使用,为了我们更好的理解caffe

自定义一个.proto文件其实是很简单,如下所示

syntax = "proto2";

package tutorial;

message Person {
  required string name = 1;
  required int32 id = 2;
  optional string email = 3;

  enum PhoneType {
    MOBILE = 0;
    HOME = 1;
    WORK = 2;
  }

  message PhoneNumber {
    required string number = 1;
    optional PhoneType type = 2 [default = HOME];
  }

  repeated PhoneNumber phones = 4;
}

message AddressBook {
  repeated Person people = 1;
}

首先是指定systax 是多少,目前有proto2 和proto3, caffe.proto 用的是proto2

然后是定义package, 这个是为了防止多个.protot 引起冲突,所以每个.proto要声明一个package,类似我们建的工程项目名

接下就是写message,message里面有三个修饰符,分别为required,optional 和 repeated

required: 这个参数必须要存在,而且只有一个

optional: 这个参数可以有也可以无,但是不超过一个

repeated: 这个参数可以重复多次

每个参数后面会有="1","2","3"... 这是为了标识这些参数,在进行编码时

当编写完.protot文件时,我们可以使用protoc 来编译

protoc -I=. --cpp_out=. ./addressbook.proto

在当前目录下会产生两个文件

如果要使用这里面的函数,将头文件include进去就可以了.