1. 程式人生 > >【16-20】視覺化:jupyter notebook做視覺化

【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、網路結構檔案與得到的模型要對應