caffe-custom new layer

来源：互联网发布：天猫淘宝客服后台操作编辑：程序博客网时间：2024/04/28 00:12

基本就是添加new_layer.cpp, new_layer.hpp, new_layer.cu三个文件, 以及修改caffe.prototxt(if the layer has parameter)：

1. new_layer.hpp

NetLayer()构造函数

LayerSetup()函数

Reshape()函数

type()层名函数

Forward_cpu()

Forward_gpu()

Backward_cpu()

Backward_gpu()

note ：这些函数都是虚函数如果再次声明的话都需要实现，所以如果没有实现就不要在这里写出来会直接执行基类的相应的函数

2. new_layer.cpp

LayerSetup()函数实现 // 初始化本层次的一些参数（一般从proto param）

Reshape()函数实现 // 根据bottom blob和一些层次参数计算得到top blob的shape

Forward_cpu() //implement bottom.data->top.data 如果需要learnable参数则是本层次的blobs_

Backward_cpu() //implement top.diff/loss -> bottom.diff 计算偏导数

LOG(INFO)<< "something debug";

3. new_layer.cu

nothing

INSTANTIATE_CLASS(NewLayer)

REGISTER_LAYER_CLASS(New)

测试方法流程，

train.prototxt (force_backwaord: true)

solver.prototxt

solver = caffe.SGDsolver(solver)

solver.net.forward( )

solver.net.blobs['newlayer'].data/diff

solver.net.params[0/1].data

https://github.com/BVLC/caffe/blob/master/examples/01-learning-lenet.ipynb

0 0