Caffe 代码学习 1: 入口及common

入口

我们先从caffe的入口main函数入手. 该函数定义在caffe.cpp中 (caffe.hpp只是包含了一堆需要用到的头文件而已). caffe提供了一系列可以调用的方法, 比如train, test, time, device_query等. 这些函数都定义在caffe.cpp里, 可以调用的函数名都注册在BrewMap g_brew_map里, 调用时通过下列函数来返回函数句柄

static BrewFunction GetBrewFunction(const caffe::string& name) 

我们具体来看一下caffe主程序提供了功能. 以caffe train为例, 入口函数会通过注册的BrewFunction找到train函数, 该函数会调用Caffe类的set_mode, SetDevice等函数, 创建solver, 并且调用solver->Solve()进行网络的训练.

common

common主要包含两部分内容: 通用的宏定义, 以及Caffe类

宏定义

这里宏定义的作用主要是对一些模板类和模板函数进行实例化. 例如下面这个宏定义我们就经常会在layer的类定义(.cpp文件)的最后看到.

// Instantiate a class with float and double specifications.
#define INSTANTIATE_CLASS(classname) \
  char gInstantiationGuard##classname; \
  template class classname; \
  template class classname

Caffe类

Caffe类的定义比较奇特, 定义了嵌套类(nested class) RNG, 以及RNG的私有嵌套类 Generator. 嵌套类的目的主要是更好的封装吧.

RNG类顾名思义, 就是Caffe的随机数产生器. 值得注意的是, Caffe类在实现时采用单例(singleton)模式, 主要通过下列几个步骤实现:

1. 禁止复制和赋值

DISABLE_COPY_AND_ASSIGN(Caffe);

2. 私有化构造函数

private:
// The private constructor to avoid duplicate instantiation.
Caffe();

3. Get方法用来获取当前实例

Caffe& Caffe::Get() {
if (!thread_instance_.get()) {
  thread_instance_.reset(new Caffe());
}
return *(thread_instance_.get());
}