import caffe
#build net
net = caffe.Net('deploy.prototxt','alexnet.caffemodel',caffe.TEST)
#list all parameters
layers_ = net.params.keys()
print layers
#list all blobs
outputs_ = net.blobs.keys()
print outputs
#extract 'conv1'
conv1_weight = net.params['conv1'][0].data
conv1_bias = net.params['conv1'][1].data
#extract output of 'conv1' layer
conv1_output = net.blobs['conv1'].data
联系客服