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