在新版本caffe里添加新的一层（向新格式中加）

最新推荐文章于 2019-03-26 14:16:48 发布

原创最新推荐文章于 2019-03-26 14:16:48 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#caffe添加新层

caffe 同时被 2 个专栏收录

22 篇文章

订阅专栏

Google库pb_glog_gflags

6 篇文章

订阅专栏

本文详细介绍了如何在新版本的Caffe中添加新的层——PatchMean层。具体步骤包括在LayerParameter中添加相应的message和ID，实现该类并添加对应的hpp和cpp文件。此方法适用于希望扩展Caffe功能的研究人员和开发者。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我前几天层写过一篇关于向新版本的caffe中添加新层的博客，那篇我已经标注了是在caffe的pb文件中使用v1LayerParameter的旧格式基础上进行的修改，我通过那篇博文的方法将新版本caffe中的euclidance loss (均方误差层)修改为patchloss层，并编译通过。而这片博文记录下：我通过仿照segmentation-caffe中unpooling添加层的方式对caffe新添加了PatchMean层，这种方式是在新版本caffe中向新格式中修改(在LayerLarameter中添加)：

1）在LayerParameter下添加相应message 和 ID

message LayerParameter {
  // ...
  optional ZZZParameter zzz_param = 143;
}

2）接着在下方添加该类的实现

// Message that stores parameters used by ZZZLayer
message ZZZParameter {
  // Whether or not slope paramters are shared across channels.
  optional int32 mul = 1 [default = 2];
}