-
初始化模型
root=’/home/*/caffe-master’
deploy=root+’examples/mnist/deply.prototxt’ //deploy位置
caffe_model=root+’examples/mnist/lenet_iter_5000.caffemodel’ //模型位置
net=caffe.Net(deploy,caffe_model,caffe.TEST) //加载model和net -
参数访问
net.params保存各层的参数值(w和b)
net.blobs保存各层的数据值
参数值:for k,v in net.params.items(): print(k) #k表示层的名称 print(v[0].data) #v表示层的参数数据 v[0].data表示w的值,v[1].data表示b的值
数据值:for s,v in net.blobs.items(): print(s) print(v.data.shape)
对于已知的层名,使用
w1=net.params[‘conv1’][0].data
#result : w1.shape为(20,1,5,5)
b1=net.params[‘conv1’][1].data