caffe学习之layer_factory(工厂方法)

最新推荐文章于 2021-02-23 15:09:02 发布

BVL10101111

最新推荐文章于 2021-02-23 15:09:02 发布

阅读量2.3k

点赞数

分类专栏： caffe 文章标签： caffe

本文链接：https://blog.csdn.net/BVL10101111/article/details/77323292

版权

本文详细介绍了Caffe中如何通过工厂方法创建layer，重点解析了LayerRegistry、REGISTER_LAYER_CLASS宏以及layer的注册过程。在Caffe中，layer的创建涉及到LayerRegistry的映射管理，以及REGISTER_LAYER_CLASS在源码中的实现。通过动态注册，网络配置文件中的layer类型能在运行时被正确创建。

摘要由CSDN通过智能技术生成

在caffe中，创建layer对象，是通过工厂方法来创建的。

相关代码，均定义在caffe/include/caffe/layer_factory.hpp，以及其实现类。

一.LayerRegistry

主要的实现是由LayerRegistry 类来完成的。
先上代码：

class LayerRegistry {
 public:
  typedef shared_ptr<Layer<Dtype> > (*Creator)(const LayerParameter&);
  typedef std::map<string, Creator> CreatorRegistry;

  static CreatorRegistry& Registry() {
    static CreatorRegistry* g_registry_ = new CreatorRegistry();
    return *g_registry_;
  }

...
...
...
}

可以看到，LayerRegistry 主要功能就是，是通过map来管理，layer的type（string）和layer的构造方法（Creator）的一个映射关系。

std::map<string, Creator>

二.REGISTER_LAYER_CLASS(type)——layer的注册

当一个xxlayer.cpp被写好之后，需要在LayerRegistry 进行相关的添加注册。

因此我们经常看到，在xxlayer.cpp的最后几行会有
REGISTER_LAYER_CLASS(type) 这行代码.

比如caffe/src/caffe/layers/data_layer.cpp中的
REGISTER_LAYER_CLASS(Data);

那么这个宏定义到底完成什么呢？具体看layer_factory中的定义。

#define REGISTER_LAYER_CLASS(type)                                             \
  template <typename Dtype>                                                    \
  s