在jupyter上顯示深度學習模型的各層
阿新 • • 發佈:2019-01-11
最近跟著網上的教程在學習深度學習,目前是用caffe框架。
參考網友的部落格(http://www.cnblogs.com/denny402/p/5103425.html),成功在jupyter上顯示出了訓練出的模型的各卷基層,但其實程式碼不太懂,先記錄到這裡,以後再分析程式碼。
基本和參考的部落格一致。過程中遇到一些問題,解釋如下:
問題1:執行出錯py:4: RuntimeWarning: invalid value encountered in divide
after removing the cwd from sys.path.
解決:plt.imshow(data)
plt.axis('off')
改為
plt.imshow(data)
plt.axis('off')
plt.show()
問題2:執行不出錯啦,但影象顯示不出來
解決:訓練模型時,我的solver的配置檔案有一行是:snapshot_format: HDF5 這使得訓練結束後模型儲存成*.h5格式,把這一行去掉後重新訓練儲存模型,就是*.caffemodel格式,再來jupyter上顯示就沒有問題了。