Caffe的解析器

Question

我正在尝试在Caffe中找到解析器。通过解析器，我指的是从文件中读取网络配置并解析它的代码部分。我想知道是否有人知道在Caffe代码库中我应该寻找这段特定的代码。

Answer 1

Caffe用于指定模型的文本文件格式使用Google Protocol Buffer格式。

您可以在src/caffe/util/io.cpp中看到读取模型的代码：

bool ReadProtoFromTextFile(const char* filename, Message* proto) {
  int fd = open(filename, O_RDONLY);
  CHECK_NE(fd, -1) << "File not found: " << filename;
  FileInputStream* input = new FileInputStream(fd);
  bool success = google::protobuf::TextFormat::Parse(input, proto);
  delete input;
  close(fd);
  return success;
}

尝试使用GitHub的搜索来查看调用此函数的代码中的位置。