caffe中的caffemodel参数提取方法
2024-04-13 20:40:09
需要的文件为：deploy.prototxt
caffemodel
net = caffe.Net(deploy.txt,caffe_model,caffe.TEST)
具体代码：
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 #提取某层数据（特征）
最新文章
- 第3月第2天 find symbolicatecrash 生产者－消费者 ice 引用计数
- salesforce 零基础开发入门学习（十）IDE便捷小知识
- x01.os.16: 添加功能
- Sql如何自动定时备份数据库
- css精灵动画
- ubuntu16.04+opencv3.1配置
- 三味书屋 bbb
- mysql主从同步单个表实验记录
- 转】Maven学习总结(五)——聚合与继承
- js常用字符串函数
- user密码
- CodeForces 203C Photographer
- 介绍一个开源的在线管理SQLServer的小工具--SQLEntMan
- 我的音乐盒子（nodejs7 + koa2 + vue + vuex + vue-router）
- Easy to Remember Color Guide for Non-Designers
- JVM核心知识体系（转http://www.cnblogs.com/wxdlut/p/10670871.html）
- 多线程线性 lock
- 关于QT应用在XP系统上兼容运行的问题
- ClouderaManager中Event Server报No such file or directory
- 01: socket模块