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 |
|
来自: 谢兴l4nztpvbdk > 《人工智能》