caffe | 添加新的层(详细过程)

一、整体流程

      针对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文件中编写注意事项。

1、.cpp和.hpp文件编写参考

------------------------------------------------------------------------参考begin-------------------------------------------------------------------------

下述.cpp和.hpp文件实现功能:

  • 3
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值