ubuntu caffe添加自己的层,smoothl1为例

1, 添加整个新层,我们以smoothl1层为例,

ssd-branch 的github 里面有现成的smoothl1 层,但官方未加入到接口里面

.cpp

.cu

.h

文件分别添加到layer里面

caffe.proto里面添加参数

注:

windows 里面是使用protoc.exe来更新生成相应的caffe.pb.h   caffe.pb.c

ubuntu的话可以使用clion,设置先编译target:proto来生成上述.pb.文件,之后再clion编译caffe  这个target  即刻

添加参数也是类似

有些细节可以参考:

https://blog.csdn.net/zh8706/article/details/95320367

 

 

 

 

 

 

你可能感兴趣的:(深度学习,编程,环境搭建)