1. 程式人生 > >在jupyter上顯示深度學習模型的各層

在jupyter上顯示深度學習模型的各層

       最近跟著網上的教程在學習深度學習,目前是用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上顯示就沒有問題了。