caffe创建Python层

红太狼 2022-05-28 03:29 265阅读 0赞

一、caffe 创建python 层

因为caffe底层是用c++编写的，所以我们有的时候想要添加某一个最新文献出来的新算法，正常的方法是直接编写c++网络层，然而这个有个前提条件是必须对caffe的底层非常熟悉，c++的编写达到一定的境界，才可灵活应用caffe，搞深度学习，走这条路对于菜鸟来说无疑很有难度。

好在caffe为我们提供了一个可以用python编写新的网络层的方法，直接用python语言编写新的网络层，然后在caffe的网络配置文件，稍作修改，就可以轻松容易创建新的网络层。其具体环境配置搭建过程如下：

1、编译caffe的时候,不能直接使用make pycaffe,而是应该使用:

**\[python\]** [view plain][] [copy][view plain]

1.  WITH\_PYTHON\_LAYER=1 make && make pycaffe

2、编写相关的网络层文件python源码filename.py,同时需要在filename.py 编写的时候,import caffe的时候,需要指向caffe所在的路径目录

3、把python源码文件filename.py拷贝到caffe/python目录下:

**\[python\]** [view plain][] [copy][view plain]

1.  caffe/python/filename.py

4、添加路径变量:

**\[python\]** [view plain][] [copy][view plain]

1.  sudo vim ~/.bashrc

在最后输入

**\[python\]** [view plain][] [copy][view plain]

1.  export PYTHONPATH=/home/hjimce/tools/caffenewversion/python:$PYTHONPATH

重启电脑.

5、调用的时候,引用层:

**\[python\]** [view plain][] [copy][view plain]

1.  layer \{ 
2.   type: 'Python'
3.   name: 'loss'
4.   top: 'loss'
5.   bottom: 'f2'
6.   bottom: 'label'
7.   python\_param \{ 
8.  \# the module name -- usually the filename -- that needs to be in $PYTHONPATH
9.   module: 'filename'
10. \# the layer name -- the class name in the module
11.  layer: 'NormlizedMSE'
12.  \} 
13. \# set loss weight so Caffe knows this is a loss layer.
14. \# since PythonLayer inherits directly from Layer, this isn't automatically
15. \# known to Caffe
16.  loss\_weight: 1
17. \}

二、caffe关闭日志输出消息

可能对于算法实验阶段来说，我们需要caffe运行网络的时候，打印网络的相关参数消息。然而对于已经走向工程之路的caffe模型来说，我们有可能需要屏蔽那些没用的输出消息，因此这个时候就需要用到caffe的消息屏蔽。

1、在程序最开始运行地方，加入如下代码：

**\[python\]** [view plain][] [copy][view plain]

1.  ::google::InitGoogleLogging("hjimce"); //初始化 
2.  FLAGS\_stderrthreshold = ::google::ERROR; //只打印ERROR级别信息

2、在程序结束的时候加入代码：

**\[python\]** [view plain][] [copy][view plain]

1.  google::ShutdownGoogleLogging();

\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*作者微博：[黄锦池-hjimce][-hjimce]博客：[http://blog.csdn.net/hjimce][http_blog.csdn.net_hjimce] 原创文章，转载请保留本行信息\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*

[view plain]: https://blog.csdn.net/hjimce/article/details/51884024?locationNum=12&fps=1#
[-hjimce]: http://weibo.com/5372176306/profile?topnav=1&wvr=6&is_all=1
[http_blog.csdn.net_hjimce]: http://blog.csdn.net/hjimce

发表评论取消回复

表情：

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

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

相关阅读

相关【caffe】关于gempool层

近期由于又重新接触到pytorch转caffe的相关任务(gempool)。发现BN并不是caffe底层代码就已经写了一个BN的实现，而是在prototxt用一些不同的算子组合

雨点打透心脏的1/2处/ 2022年09月16日 04:59/ 0 赞/ 233 阅读

相关 Caffe 基本层参数简介

Caffe更新很快，以下内容可能与最新master有出入，不过如果参数名字没有更改应该没有变化。具体可以查看src/caffe/proto/caffe.proto文件内的定义。

布满荆棘的人生/ 2022年08月04日 05:25/ 0 赞/ 229 阅读

相关 caffe中各层定义

![Center][] [Vision Layers][] [Convolution][] [Pooling][]

深碍√TFBOYSˉ_/ 2022年07月14日 03:21/ 0 赞/ 244 阅读

相关 Caffe 基本层参数简介

Caffe更新很快，以下内容可能与最新master有出入，不过如果参数名字没有更改应该没有变化。具体可以查看src/caffe/proto/caffe.proto文件内的定义。

桃扇骨/ 2022年05月29日 06:23/ 0 赞/ 237 阅读

相关 caffe创建Python层

一、caffe 创建python 层因为caffe底层是用c++编写的，所以我们有的时候想要添加某一个最新文献出来的新算法，正常的方法是直接编写c++网络层，然而这个有个前

红太狼/ 2022年05月28日 03:29/ 0 赞/ 266 阅读

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

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

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

相关 caffe添加自定义python层

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

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

相关 caffe之(二)pooling层

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

谁践踏了优雅/ 2021年07月17日 00:04/ 0 赞/ 462 阅读

相关 caffe之(五)loss层

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

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

相关 caffe之（三）激活函数层

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

古城微笑少年丶/ 2021年07月16日 12:58/ 0 赞/ 413 阅读