分享

pycaffe中权值(params)以及各个层的输出(blobs)的获取 | 21世纪郭少加油

 谢兴l4nztpvbdk 2017-08-30
  • 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
    • 本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
      转藏 分享 献花(0

      0条评论

      发表

      请遵守用户 评论公约

      类似文章 更多