caffe新建layer完整流程;

Posted 物物而不物于wu

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了caffe新建layer完整流程;相关的知识,希望对你有一定的参考价值。

caffe 新建layer的完整流程(以添加sin_layer为例)

一般,新建layer分为cpu版本和gpu版本;

一、CPU版本

分为以下步骤:

  1. 在./include/caffe/layer文件下新建sin_layer.hpp文件;
  2. 在./src/caffe/layers文件下新建sin_layer.cpp文件;
  3. 在./src/caffe/proto/caffe.proto文件中新增消息名;
  4. 在./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完整流程;的主要内容,如果未能解决你的问题,请参考以下文章

Caffe 源码阅读 卷积层

一文带你熟悉Pytorch-;Caffe-;om模型转换流程

在caffe中添加新的layer

如何给caffe添加新的layer ?

Caffe : Layer Catalogue

Caffe : Layer Catalogue