博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
查看caffemodel的参数值
阅读量:5010 次
发布时间:2019-06-12

本文共 1281 字,大约阅读时间需要 4 分钟。

查看参数值的代码

#-*-coding:utf-8-*-import caffeimport sysimport numpy as npimport getoptimport re   if __name__ =="__main__":	opts,args = getopt.getopt(sys.argv[1:],"m:w:")	MODEL_FILE = "test-13.prototxt"	PRETRAIN_FILE = "n36_faster_rcnn__binarizedmodel_svd_fc6_512_fc7_512.caffemodel"	#process	params_txt = 'params.txt'	pf = open(params_txt, 'w')	print MODEL_FILE	print PRETRAIN_FILE        net = caffe.Net (MODEL_FILE, PRETRAIN_FILE, caffe.TEST)        for param_name in net.params.keys():		if re.match('conv.',param_name) or re.match('fc.',param_name):			print 'zhaodezan'                	weight = net.params[param_name][0].data                	bias = net.params[param_name][1].data               		 #保存开始                	pf.write(param_name)                	pf.write('\n')               		 #保存weight                	pf.write('\n' + param_name + '_weight:\n\n')                	weight.shape = (-1, 1)                	for w in weight:                        	pf.write('%ff, ' % w)                	#保存bias                	pf.write('\n\n' + param_name + '_bias:\n\n')               		bias.shape = (-1, 1)                	for b in bias:                        	pf.write('%ff, ' % b)                	pf.write('\n\n')        pf.close

转载于:https://www.cnblogs.com/hellokittyblog/p/9128408.html

你可能感兴趣的文章
ue4同c#通信时的中文乱码问题
查看>>
黄老师架构师课程笔记(二)
查看>>
mvc性能优化
查看>>
log
查看>>
663 如何做“低端”产品?(如何把低端做得高端 - 认同感)
查看>>
JDBC 第九课 —— 初次接触 JUnit
查看>>
Windows核心编程:第10章 同步设备IO与异步设备IO
查看>>
浏览器加载、解析、渲染的过程
查看>>
开放api接口签名验证
查看>>
sed 常用操作纪实
查看>>
C++复习:对C的拓展
查看>>
校外实习报告(九)
查看>>
android之android.intent.category.DEFAULT的用途和使用
查看>>
CAGradientLayer 透明渐变注意地方(原创)
查看>>
织梦DEDE多选项筛选_联动筛选功能的实现_二次开发
查看>>
iOS关于RunLoop和Timer
查看>>
SQL处理层次型数据的策略对比:Adjacency list vs. nested sets: MySQL【转载】
查看>>
已存在同名的数据库,或指定的文件无法打开或位于 UNC 共享目录中。
查看>>
MySQL的随机数函数rand()的使用技巧
查看>>
thymeleaf+bootstrap,onclick传参实现模态框中遇到的错误
查看>>