caffe下添加新的loss层

悠悠 2022-05-20 00:21 245阅读 0赞

caffe的使用当中，最神秘的就是caffe的源码，而我们要做一些创新点的话，基本上都是要修改caffe的源码。  
小鱼和大家分享怎么在caffe下添加新的层。  
比如要添加一个新的loss函数：euclidean\_loss\_norm\_layer

1、需要编写相应的euclidean\_loss\_norm\_layer.cpp和相应的euclidean\_loss\_norm\_layer.cu文件，.cpp文件是实现cpu下函数的前传与反传，.cu文件是实现gpu下函数的前传与反传；  
2、将euclidean\_loss\_norm\_layer.cpp和  
euclidean\_loss\_norm\_layer.cu文件放入caffe/src/caffe/layers文件夹下；  
3、编写对应的euclidean\_loss\_norm\_layer.hpp文件，放入caffe/include/caffe/layers文件夹下；  
4、在caffe/src/caffe/proto/caffe.proto文件下添加对应的optional及meaasge，如下：

optional EuclideanLossNormParameter euclidean_loss_norm_param = 149;

message EuclideanLossNormParameter {
      // The axis along which to concatenate -- may be negative to index from the
      // end (e.g., -1 for the last axis).  Other axes must have the
      // same dimension for all the bottom blobs.
      // By default, ConcatLayer concatenates blobs along the "channels" axis (1).
     // optional int32 axis = 2 [default = 1];
    
      // DEPRECATED: alias for "axis" -- does not support negative indexing.
      //optional uint32 concat_dim = 1 [default = 1];
    }

5、重新在caffe下make clean及make all，编译通过即可。

如果喜欢请点个赞再走哦，有什么问题请留言区讨论，一起学习。

发表评论取消回复

表情：

评论列表（有 0 条评论，245人围观）

还没有评论，来说两句吧...

相关阅读

相关为caffe添加新的层

[how to develop new layers][] [caffe中增加自己的layer][caffe_layer] [【深夜福利

野性酷女/ 2022年07月13日 09:11/ 0 赞/ 244 阅读

相关 triplet loss的原理及caffe代码

1：triplet loss的原理及梯度推到 [http://blog.csdn.net/tangwei2014/article/details/46788025][http_

心已赠人/ 2022年07月13日 07:56/ 0 赞/ 182 阅读

相关 caffe画loss曲线

[用图形画出caffe输出数据的python程序&&git基础命令][caffe_python_git] [caffe 画损失曲线和准确率曲线

以你之姓@/ 2022年07月13日 05:28/ 0 赞/ 188 阅读

相关 caffe工具绘制 loss accuracy曲线

深度学习中loss和accuracy是十分重要的数据 …balabala 这篇博主根据[http://blog.csdn.net/u013078356/article/de

男娘i/ 2022年06月05日 07:25/ 0 赞/ 210 阅读

相关 caffe下添加新的loss层

caffe的使用当中，最神秘的就是caffe的源码，而我们要做一些创新点的话，基本上都是要修改caffe的源码。小鱼和大家分享怎么在caffe下添加新的层。比如要添

悠悠/ 2022年05月20日 00:21/ 0 赞/ 246 阅读

相关 Caffe添加Center_Loss layer

1. Caffe修改 1.1 修改caffe.proto 在message LayerParameter\{\}中添加如下代码： optional Cen

「爱情、让人受尽委屈。」/ 2022年05月11日 04:14/ 0 赞/ 213 阅读

相关 caffe中的各种loss函数

转自：[https://blog.csdn.net/u012177034/article/details/52144325][https_blog.csdn.net_u0121

约定不等于承诺〃/ 2022年05月11日 00:00/ 0 赞/ 233 阅读

相关 caffe添加自己编写的Python层

1. 前言由于Python的灵活性，我们在caffe中添加自己定义的层时使用python层会更加方便，开发速速也会比C++更快，要使用python层需要编译caffe的

太过爱你忘了你带给我的痛/ 2022年02月23日 07:58/ 0 赞/ 218 阅读

相关 caffe添加自定义python层

参考文章： [https://blog.csdn.net/qq\_41648043/article/details/82556017][https_blog.csdn.ne

喜欢ヅ旅行/ 2022年02月04日 00:25/ 0 赞/ 282 阅读

相关 caffe之(五)loss层

在caffe中，网络的结构由prototxt文件中给出，由一些列的Layer（层）组成，常用的层如：数据加载层、卷积操作层、pooling层、非线性变换层、内积运算层、归一化层

末蓝、/ 2021年07月16日 23:53/ 0 赞/ 357 阅读