查看caffemodel的参数值

来源：互联网发布：中国巴基斯坦贸易数据编辑：程序博客网时间：2024/06/06 01:19

查看参数值的代码

#-*-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"#processparams_txt = 'params.txt'pf = open(params_txt, 'w')print MODEL_FILEprint 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

阅读全文

0 0