Caffe小玩意(2)-从caffemodel中导出参数
来源:互联网 发布:北京文森特软件科技 编辑:程序博客网 时间:2024/05/16 04:09
Caffe小玩意(2)-从caffemodel中导出参数
import numpy as npimport scipy.io as sioimport caffedef load(): # Load the net caffe.set_mode_cpu() # You may need to train this caffemodel first # There should be script to help you do the training net = caffe.Net(root + 'lenet.prototxt', root + 'lenet_iter_10000.caffemodel',\ caffe.TEST) conv1_w = net.params['conv1'][0].data conv1_b = net.params['conv1'][1].data conv2_w = net.params['conv2'][0].data conv2_b = net.params['conv2'][1].data ip1_w = net.params['ip1'][0].data ip1_b = net.params['ip1'][1].data ip2_w = net.params['ip2'][0].data ip2_b = net.params['ip2'][1].data sio.savemat('conv1_w', {'conv1_w':conv1_w}) sio.savemat('conv1_b', {'conv1_b':conv1_b}) sio.savemat('conv2_w', {'conv2_w':conv2_w}) sio.savemat('conv2_b', {'conv2_b':conv2_b}) sio.savemat('ip1_w', {'ip1_w':ip1_w}) sio.savemat('ip1_b', {'ip1_b':ip1_b}) sio.savemat('ip2_w', {'ip2_w':ip2_w}) sio.savemat('ip2_b', {'ip2_b':ip2_b})if __name__ == "__main__": # You will need to change this path root = '/Users/yuliangzou/caffe-rc3/examples/mnist/' load() print 'Caffemodel loaded and written to .mat files successfully!'
从代码里可以看得很清楚啦,首先导入模型,然后利用net.params就可以获取参数了,另外你也可以利用net.data导出数据进行可视化。当然,在导出参数之前…你必须要跑过一遍,不然你没有这个caffemodel…
https://github.com/Yuliang-Zou/Naive-CNN
0 0
- Caffe小玩意(2)-从caffemodel中导出参数
- 从caffemodel中导出参数
- 从caffemodel中导出参数
- caffe上手:如何导出caffemodel参数
- caffe上手:如何导出caffemodel参数
- 如何导出caffemodel参数
- Caffe小玩意
- Caffe学习:使用pycaffe读取caffemodel参数
- Caffe学习:使用pycaffe读取caffemodel参数
- Caffe学习:使用pycaffe读取caffemodel参数
- Caffe学习:使用pycaffe读取caffemodel参数
- Caffe小玩意(1)-可视化网络结构
- Caffe小玩意(1)-可视化网络结构
- caffe中 对caffemodel进行结构修改并保存
- 小玩意
- 小玩意
- Caffe学习:pycaffe利用caffemodel进行分类(2)
- Caffe学习:pycaffe利用caffemodel进行分类(2)
- CF Round #358 D
- leecode_324 Wiggle Sort II
- State bank of India Shimla Himachal Pradesh Ifsc Codes
- Android 播放视频并获取指定时间的帧画面
- 整理Ruby相关的各种概念(rvm, gem, bundle, rake, rails等)
- Caffe小玩意(2)-从caffemodel中导出参数
- 运维中的一些概念scale-in/scale-out/scale-up/scale-down/auto-scaling
- Maximum Product Subarray
- leetcode #105 in cpp
- 简单的java代码在内存加载过程(详解)
- android 获取网络视频缩略图
- Android之使用MediaMetadataRetriever类获取视频第一帧及用法
- leetcode #106 in cpp
- 日常水