caffe 添加YOLO新层Leaky Layer

1.创建.hpp头文件leaky_layer.hpp,放入 F:\caffe-master\include\caffe\layers
2.创建.cpp头文件leaky_layer.cpp,放入 F:\caffe-master\src\caffe\layers

4.修改src\caffe\proto\caffe.proto文件,修改两处地方:
(1)
//第一处下面添加我们的leaky层的参数 取id为 151
optional LEAKYParameter leaky_param = 151;//leaky层的总参数名
(2)
//1000行。第二处leaky层的参数
message LEAKYParameter {
optional float leaky_scale = 1 [default = 0.1]; //默认为0.1
}

5.修改 F:\caffe-master\windows\libcaffe 下的两个文件
libcaffe.vcxproj 与 libcaffe.vcxproj.filters
notepad打开libcaffe.vcxproj ,在165行附近加:
这里写图片描述
上图为cpp的路径,下面在libcaffe.vcxproj.filters中添加hpp文件路径:
这里写图片描述

6.修改完删掉caffe-windows\src\caffe\proto\caffe.pb.cc文件和caffe-windows\include\caffe\proto\caffe.pb.h文件
7.编译时 要先编译libcaffe 然后编译caffe 再编译其他tools
8.编译好之后可以用MNIST数据集测试一下,修改caffe/example/mnist/lenet_train_test.prototxt中:
layer {
name: “Leaky1”
type: “LEAKY”
bottom: “ip1”
top: “ip1”
}
也就是吧relu改为我们刚新建的LEAKY。
9.保存后,打开CMD,caffe路径下运行下面的指令:

可以看到,运行正常。
这里写图片描述
并且可以看到我们新建的leaky层确实用到了 
这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值