如何在Caffe中加入新的layer并利用Matlab进行调试
Caffe当中已经实现了很多常用的layer,但是在有些时候,通过单独使用或者组合这些已有的layer并没法实现我们想要的功能,这时便只能自己添加一个新的layer了,最新版Caffe添加一个layer主要步骤有四个:
- 在proto中添加layer相应的参数
- 编写实现layer的函数:hpp、cpp、cu
- 有些layer需要在layer_factory中进行装配
- layer的调试
下面以实现一个激活函数层为例,来完整地进行这样一个步骤,并用Matlab来进行调试。
我们要实现如下的激活函数:
f(x)=⎧⎩⎨⎪⎪x−10