首页 > 其他 > 详细

Caffe学习系列(15):添加新层

时间:2016-11-19 23:49:38      阅读:319      评论:0      收藏:0      [点我收藏+]

如何在Caffe中增加一层新的Layer呢?主要分为四步:

(1)在./src/caffe/proto/caffe.proto 中增加对应layer的paramter message;

(2)在./include/caffe/***layers.hpp中增加该layer的类的声明,***表示有common_layers.hpp,

data_layers.hpp, neuron_layers.hpp, vision_layers.hpp 和loss_layers.hpp等;

(3)在./src/caffe/layers/目录下新建.cpp和.cu(GPU)文件,进行类实现。

(4)在./src/caffe/gtest/中增加layer的测试代码,对所写的layer前传和反传进行测试,测试还包括速度。(可省略,但建议加上)

  

  这位博主添加了一个计算梯度的网络层,简介明了:

  http://blog.csdn.net/shuzfan/article/details/51322976

 

  这几位博主增加了自定义的loss层,可供参考:

 

  http://blog.csdn.net/langb2014/article/details/50489305

 

  http://blog.csdn.net/tangwei2014/article/details/46815231

 

Caffe学习系列(15):添加新层

原文:http://www.cnblogs.com/573177885qq/p/6065625.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!