caffe 网络结构和模型权重文件.caffemodel可视化
1、caffe定义网络结构是用protobuf格式写的.prototxt文件,如果我们想对其可视化,可以使用Netscope对其可视化
(1)打开http://ethereon.github.io/netscope/#/editor
(2)将.prototxt文件输入到左边文本框,即可可视化
2.通过代码可以读取.caffemodel参数与结构
代码如下:
#先要将自己的PYTHONPATH指定到caffe的python路径
import caffe
model_path = '<path to caffemodel>'
ptototxt_path = '<path to prototxt>'
caffe.set_mode_gpu()
net = caffe.Net(prototxt, model, caffe.TEST)
此时网络加载如图:
后面可以通过net.params[l].[0].data查看w权重值,l代表当前层的名字