为caffe添加ReLU6支持

前一篇文章已经安装了caffe,不过由于要用到的SSDLite需要ReLU6的支持,因此需要修改caffe部分代码以及添加几个文件。


 首先找到src/caffe/proto/caffe.proto这个文件,找到message LayerParameter后在其内的optional ReLUParameter relu_param = 123;之后添加optional ReLU6Parameter relu6_param = 208;

 然后再找到message ReLUParameter并在其后添加

// Message that stores parameters used by ReLU6Layer
message ReLU6Parameter {
  enum Engine {
    DEFAULT = 0;
    CAFFE = 1;
    CUDNN = 2;
  }
  optional Engine engine = 2 [default = DEFAULT];
}

 修改完毕,接下来添加所需的头文件和实现文件。

 在include/caffe/layers路径下添加cudnn_relu6_layer.hpprelu6_layer.hpp

 在src/caffe/layers路径下添加cudnn_relu6_layer.cpprelu6_layer.cppcudnn_relu6_layer.cu以及relu6_layer.cu

 由于作了修改,所以需要重新编译,这里就直接重新创建了新环境,然后按照上一篇文章的方式再次编译安装caffe。

  • 3
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值