怎么提取caffe训练好的model：.caffemodel中的数据 30
1个回答
展开全部
以下是 python 代码，只描述了大致思想
import scipy.io as sio
net = caffe.Net(prototxt, caffemodel, caffe.TEST)
for layer in net.key:
weight = net.param[layer][0] # 存放 weight
bias = net.param[layer][1] # 存放 bias
sio.savemat (output_file_name, {'layer': layer, 'weight': weight, 'bias': bias})
以上
import scipy.io as sio
net = caffe.Net(prototxt, caffemodel, caffe.TEST)
for layer in net.key:
weight = net.param[layer][0] # 存放 weight
bias = net.param[layer][1] # 存放 bias
sio.savemat (output_file_name, {'layer': layer, 'weight': weight, 'bias': bias})
以上
本回答由网友推荐
已赞过
已踩过<
评论
收起
你对这个回答的评价是？
推荐律师服务：
若未解决您的问题，请您详细描述您的问题，通过百度律临进行免费专业咨询
