CAFFE学习笔记(二)Caffe_Example之测试mnist
这一次的博客将接着上一次的内容,简单讲解一下如何使用训练后的网络lenet_iter_5000.caffemodel与lenet_iter_10000.caffemodel。
1、在网络训练完毕后,将在指定路径后生成四个文件,其中两个caffemodel文件lenet_iter_5000.caffemodel与lenet_iter_10000.caffemodel为我们测试时所需要的,它们分别存放着程序迭代5000次与10000次后网络的参数。
为了防止连自己都不知道这两个文件将在哪里生成,我们首先需要在训练之前就先设置好caffemodel文件的生成路径,如图:
然后接下来等待网络训练完毕后,将在对应的路径下生成四个文件:
2、在CAFFE/examples/mnist目录下新建一个文本文件,然后将后缀名改成sh。填入以下语句:
注意,这样填写只是为了便于观看,但直接保存后是不能跑起来的,必须把第2~5条语句全部放在同一行,语句与语句之间用一个空格隔开。
3、保存该sh文件,然后双击,即可出现训练情况:
2016.5.7
by 悠望南山