Caffe源码中添加新层 详细教程

来源:互联网 发布:学matlabgui编程的意义 编辑:程序博客网 时间:2024/06/08 08:01

Caffe源码中添加新层 详细教程

增加一个AndyLayer,作用和ConvolutionLayer一样,只是用来说明添加新层的步骤以及测试添加的新层的步骤。

1.在 include/caffe/vision_layers.hpp 中添加 andylayer的定义

2.将andy_layer.cpp添加到src\caffe\layers文件夹中,代码内容复制convolutio_layer.cpp即可

3.  修改proto/caffe.proto文件,找到LayerType,添加ANDY,并更新ID。假如说Andy_Layer有参数,比如Convolution肯定是有参数的,那么添加AndyParameter类,仿照Convolution进行修改。




0 0