怎么提取caffe训练好的model:.caffemodel中的数据

2025-04-26 05:31:54
推荐回答(1个)
回答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})

以上