在jupyter上显示深度学习模型的各层

       最近跟着网上的教程在学习深度学习,目前是用caffe框架。

       参考网友的博客(http://www.cnblogs.com/denny402/p/5103425.html),成功在jupyter上显示出了训练出的模型的各卷基层,但其实代码不太懂,先记录到这里,以后再分析代码。

在jupyter上显示深度学习模型的各层_第1张图片

在jupyter上显示深度学习模型的各层_第2张图片

在jupyter上显示深度学习模型的各层_第3张图片

在jupyter上显示深度学习模型的各层_第4张图片

    基本和参考的博客一致。过程中遇到一些问题,解释如下:

    问题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上显示就没有问题了。

       

你可能感兴趣的:(caffe)