一、整体流程
针对caffe添加新的层,网上有许多教程,这里我们举一例子,假设我想实现如下功能:将一个wxhxc的特征图reshape成(w/sample_step) x (h/smple_step) x (c*sample_step*sample_step),也就是实现简单的每隔sample_step点下采,然后把所有的通道再拼起来,即channel的维度提高sample_step*sample_step倍。假设我要添加的新层名字是FeatReshapeCommon,我们需要准备如下几个文件:一个是要添加的新层的三个文件feat_reshape_common_layer.cpp、feat_reshape_common_layer.hpp、feat_reshape_common_layer.cu.
注意:
1.这里的三个文件名字要与我们需要添加的名字保持一致,即层名字FeatReshapeCommon,新层的三个文件feat_reshape_common_layer.cpp、feat_reshape_common_layer.hpp、feat_reshape_common_layer.cu。
2、第三个cu文件是实现GPU上前向传播(Forward_gpu),反向传播(Backward_gpu)进行并行加速的,不是必须选项,如果对CUDA编程不熟悉可以不实现第三个feat_reshape_common_layer.cu函数,也就是不往caffe_root/src/caffe/layers/下添加.cu文件即可。因为我们会在feat_reshape_common_layer.cpp中实现cpu版本的前向传播(Forward_cpu),反向传播(Backward_cpu)函数。
二、编写上述三个文件
关于feat_reshape_common_layer.cpp、feat_reshape_common_layer.hpp具体的编写过程请参考
caffe中添加新层(差分层)博客,感谢这篇博客内容的原著者,这里楼主把其中的内容内容拿过来,方便查阅,后面楼主主要讲下 .cu文件中编写注意事项。
------------------------------------------------------------------------参考begin-------------------------------------------------------------------------
下述.cpp和.hpp文件实现功能: