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 的搜索来查看调用此函数的代码中的位置。

Caffe 中的解析器

Parser in Caffe

neural-network

caffe