caffe的python接口caffemodel参数及特征抽取示例
正文
如果用公式 y=f(wx+b)
来表示整个运算过程的话，那么w和b就是我们需要训练的东西，w称为权值，在cnn中也可以叫做卷积核（filter），b是偏置项。f是激活函数，有sigmoid、relu等。x就是输入的数据。
数据训练完成后，保存的caffemodel里面，实际上就是各层的w和b值。
我们运行代码：
deploy=root + 'mnist/deploy.prototxt' #deploy文件 caffe_model=root + 'mnist/lenet_iter_9380.caffemodel' #训练好的 caffemodel net = caffe.Net(net_file,caffe_model,caffe.TEST) #加载model和network
就把所有的参数和数据都加载到一个net变量里面了，但是net是一个很复杂的object, 想直接显示出来看是不行的。其中：
net.params: 保存各层的参数值（w和b)
net.blobs: 保存各层的数据值
可用命令：
[(k,v[0].data) for k,v in net.params.items()]
查看各层的参数值，其中k表示层的名称，v[0].data就是各层的W值，而v[1].data是各层的b值。注意：并不是所有的层都有参数，只有卷积层和全连接层才有。
也可以不查看具体值，只想看一下shape,可用命令
[(k,v[0].data.shape) for k,v in net.params.items()]
假设我们知道其中第一个卷积层的名字叫'Convolution1', 则我们可以提取这个层的参数：
w1=net.params['Convolution1'][0].data b1=net.params['Convolution1'][1].data
输入这些代码，实际查看一下，对你理解network非常有帮助。
同理，除了查看参数，我们还可以查看数据，但是要注意的是，net里面刚开始是没有数据的，需要运行：
net.forward()
之后才会有数据。我们可以用代码：
[(k,v.data.shape) for k,v in net.blobs.items()]
或
[(k,v.data) for k,v in net.blobs.items()]
来查看各层的数据。注意和上面查看参数的区别，一个是net.params, 一个是net.blobs.
实际上数据刚输入的时候，我们叫图片数据，卷积之后我们就叫特征了。
如果要抽取第一个全连接层的特征，则可用命令：
fea=net.blobs['InnerProduct1'].data
只要知道某个层的名称，就可以抽取这个层的特征。
推荐大家在spyder中，运行一下上面的所有代码，深入理解模型各层。
最后，总结一个代码：
import caffe import numpy as np root='/home/xxx/' #根目录 deploy=root + 'mnist/deploy.prototxt' #deploy文件 caffe_model=root + 'mnist/lenet_iter_9380.caffemodel' #训练好的 caffemodel net = caffe.Net(deploy,caffe_model,caffe.TEST) #加载model和network [(k,v[0].data.shape) for k,v in net.params.items()] #查看各层参数规模 w1=net.params['Convolution1'][0].data #提取参数w b1=net.params['Convolution1'][1].data #提取参数b net.forward() #运行测试 [(k,v.data.shape) for k,v in net.blobs.items()] #查看各层数据规模 fea=net.blobs['InnerProduct1'].data #提取某层数据（特征）
以上就是caffe的python接口caffemodel参数及特征抽取的详细内容，更多关于python caffemodel参数特征抽取的资料请关注脚本之家其它相关文章！
相关文章
今天抽空给大家介绍Django模板层过滤器和继承的问题，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友参考下吧2021-07-07
这篇文章主要介绍了使用cx_freeze把python打包成exe的示例，大家参考使用吧2014-01-01
这篇文章是入门级别的应用Python + Selenium进行自动化测试，包括环境搭建及简单的实例，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-09-09
这篇文章主要介绍了解决django 向mysql中写入中文字符出错的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
Pandas是数据科学中的利器,你可能想到的数据处理骚操作,貌似用Pandas都能够实现,下面这篇文章主要给大家介绍了关于python pandas数据处理之删除特定行与列的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-08-08
要在IDEA开发环境中配置Python程序,你需要先安装Python,可以从官网上下载Python的安装包,按照提示进行安装即可,下面这篇文章主要给大家介绍了关于IDEA安装python插件并配置的详细图文教程,需要的朋友可以参考下2024-03-03
Python实现csv文件(点表和线表)转换为shapefile文件的方法
这篇文章主要介绍了Python实现csv文件(点表和线表)转换为shapefile文件的方法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-10-10
这篇文章主要为大家详细介绍了python实现键盘控制鼠标移动，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-10-10
这篇文章主要介绍了Python实现桶排序与快速排序算法结合应用,结合实例形式分析了Python快速排序及桶排序结合应用的相关实现技巧,需要的朋友可以参考下2017-11-11
Flask真是一个强大且简介的web框架，能够快速搭建web服务器，本文主要介绍了Flask搭建一个API服务器的步骤，分享给大家，感兴趣的可以了解一下2021-05-05
最新评论