您的位置:首页 > 其它

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

2017-06-28 20:09 351 查看
       最近跟着网上的教程在学习深度学习,目前是用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上显示就没有问题了。

       
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: