使用pycaffe读取caffemodel参数（保存到txt文件）

#!/usr/bin/env python
# 引入“咖啡”
import caffe
import numpy as np
# 使输出的参数完全显示
# 若没有这一句，因为参数太多，中间会以省略号“……”的形式代替
np.set_printoptions(threshold='nan')
# deploy文件
MODEL_FILE = 'caffe_deploy.prototxt'
# 预先训练好的caffe模型
PRETRAIN_FILE = 'caffe_iter_10000.caffemodel'
# 保存参数的文件
params_txt = 'params.txt'
pf = open(params_txt, 'w')
# 让caffe以测试模式读取网络参数
net = caffe.Net(MODEL_FILE, PRETRAIN_FILE, caffe.TEST)
# 遍历每一层
for param_name in net.params.keys():
# 权重参数
weight = net.params[param_name][0].data
# 偏置参数
bias = net.params[param_name][1].data
# 该层在prototxt文件中对应“top”的名称
pf.write(param_name)
pf.write('\n')
# 写权重参数
pf.write('\n' + param_name + '_weight:\n\n')
# 权重参数是多维数组，为了方便输出，转为单列数组
weight.shape = (-1, 1)
for w in weight:
pf.write('%ff, ' % w)
# 写偏置参数
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

使用pycaffe读取caffemodel参数（保存到txt文件）相关推荐

python将二维和三维数组保存到txt文件
首先要保证在你的python环境中装了numpy包,接下来我们要使用其中的两个函数用于存储和加载数组:np.savetxt()和np.loadtxt(). 二维数组的保存和获取 import nump ...
在当当买了python怎么下载源代码-初学Python 之抓取当当网图书页面目录并保存到txt文件...
这学期新开了门"高大上"的课<机器学习>,也开始入门Python.然后跟我们一样初学Python 的老师布置了个"作业"--用Python 弄个抓取 ...
np.savetxt()——将array保存到txt文件，并保持原格式
np.savetxt()--将array保存到txt文件,并保持原格式 **问题:**1.如何将array保存到txt文件中?2.如何将存到txt文件中的数据读出为ndarray类型? **需求:** ...
python输出print到txt-pycharm实现print输出保存到txt文件
程序比较大,运行时间又长的时候,不想等着结果,只要先确保程序功能正确无误,可以把需要的结果print然后保存到TXT文件中,超级方便 # 创建一个txt文件,文件名为mytxtfile def tex ...
C语言下链表数据保存到.txt文件失败
问题描述保存链表中的数据的代码 int readin3(ConRecNode* head) {ListConRec p = head->next;FILE* fp;char fileName[ ...
python 爬取HTML内容并保存到txt文件内
# @UpdateTime : 2020-12-08 16:53 # @Author : wz # @File : Get_WebDetails # @Software: PyCharm # @use ...
ffmpeg读取rtsp并保存到mp4文件
本文章只讲述mp4文件的录像至于音频录入会在下个文章中介绍总体思路为:初始化----连接相机获取码流--读取码流中的视频--创建输出mp4上下文---写mp4头----循环读取码流--写入mp4 ...
146_路由器项目_附加功能：可以拿取文件内IP保存到过滤链表中，过滤IP也可以保存到txt文件，也可以终端输入IP，保存到文件【单链表的尾插、头插、删除指定节点、遍历等】】
动图: ip_file.h: #ifndef IP_FILE_H #define IP_FILE_H #include <stdio.h> #include <string.h> ...
Scrapy保存到txt文件或者数据库里面
首先要理解yield的作用. 在pipelines.py文件中定义了保存方法,但是一直没有保存,为什么呢? file = open("E:\\Scripy\\spidersitems.txt ...

使用pycaffe读取caffemodel参数（保存到txt文件）

使用pycaffe读取caffemodel参数（保存到txt文件）相关推荐

最新文章

热门文章