【16-20】可视化:jupyter notebook做可视化

由于使用的是jupyter notebook,这里不做展示,详细内容可参考github

net = caffe.Net(caffe_root + 'examples/cifar10/cifar10_full.prototxt',
                caffe_root + 'examples/cifar10/my_iter_10000.caffemodel',
                caffe.TEST)
[(k, v[0].data.shape) for k, v in net.params.items()]

这里主要有一个问题是执行到这里,jupyter notebook总是崩溃,按照博主的说法是因为两点:

1、需要的是网络部署文件,即有网络结构的那个文件,而不是配置的文件

2、网络结构文件与得到的模型要对应

猜你喜欢

转载自blog.csdn.net/yeler082/article/details/83097642