caffe中添加新层的步骤

caffe中添加新层的步骤
1、增加caffe/include/caffe/layers/.hpp文件
2、增加caffe/src/caffe/layers/.cpp文件,若要用GPU,还要写.cu文件
3、修改src/caffe/proto/caffe.proto文件:改两个地方,如下

3.1一个是在message LayerParameter {}中添加一个唯一ID号,注意不要重复;
3.2另一个是在后面的任意位置处定义增加的这个层的参数,比如像message InnerProductParameter {}这些这样

4、重新编译caffe的步骤:

make clean;
make all -j4 ;
make test ;
make runtest ;
make pycaffe

猜你喜欢

转载自blog.csdn.net/blankai/article/details/80077226