caffe新建layer完整流程;
Posted 物物而不物于wu
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了caffe新建layer完整流程;相关的知识,希望对你有一定的参考价值。
caffe 新建layer的完整流程(以添加sin_layer为例)
一般,新建layer分为cpu版本和gpu版本;
一、CPU版本
分为以下步骤:
- 在./include/caffe/layer文件下新建sin_layer.hpp文件;
- 在./src/caffe/layers文件下新建sin_layer.cpp文件;
- 在./src/caffe/proto/caffe.proto文件中新增消息名;
- 在./src/caffe/alyers文件下新建test_sin_layer.cpp文件进行单元测试;
1. sin_layer.hpp文件
在sin_layer.hpp文件下添加如下代码:
#ifndef CAFFE_SIN_LAYER_HPP_
#define CAFFE_SIN_LAYER_HPP_
以上是关于caffe新建layer完整流程;的主要内容,如果未能解决你的问题,请参考以下文章