dump_caffemodel_weights.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:seglink 作者: bgshih 项目源码 文件源码
def dump_caffemodel_weights():
  net = caffe.Net(args.prototxt_path, args.caffemodel_path, caffe.TEST)
  weights = {}
  n_layers = len(net.layers)
  for i in range(n_layers):
    layer_name = net._layer_names[i]
    layer = net.layers[i]
    layer_blobs = [o.data for o in layer.blobs]
    weights[layer_name] = layer_blobs
  joblib.dump(weights, args.caffe_weights_path)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号