caffe中的caffemodel參數提取方法


需要的文件為:deploy.prototxt

                         caffemodel

net = caffe.Net(deploy.txt,caffe_model,caffe.TEST)
具體代碼:

import caffe
import numpy as np
root='/home/xxx/' #根目錄
deploy=root + 'mnist/deploy.prototxt' #deploy文件
caffe_model=root + 'mnist/lenet_iter_9380.caffemodel' #訓練好的 caffemodel
net = caffe.Net(deploy,caffe_model,caffe.TEST) #加載model和network

[(k,v[0].data.shape) for k,v in net.params.items()] #查看各層參數規模
w1=net.params['Convolution1'][0].data #提取參數w
b1=net.params['Convolution1'][1].data #提取參數b

net.forward() #運行測試

[(k,v.data.shape) for k,v in net.blobs.items()] #查看各層數據規模
fea=net.blobs['InnerProduct1'].data #提取某層數據(特征)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM