将Caffe的环境搭好了,现在用MNIST这个数据集进行测试,继续在caffe下进行操作
1.输入./data/mnist/get_mnist.sh
,会在mnIst文件夹下看到四个压缩包,解压之后放在此文件夹下
2.在caffe的终端命令下输入
./examples/mnist/create_mnist.sh
会出现如下错误
F0121 20:00:19.648351 3013 convert_mnist_data.cpp:48] Check failed: image_file Unable to open file data/mnist/train-images-idx3-ubyte
*** Check failure stack trace: ***
@ 0x7f40a0b8edaa (unknown)
@ 0x7f40a0b8ece4 (unknown)
@ 0x7f40a0b8e6e6 (unknown)
@ 0x7f40a0b91687 (unknown)
@ 0x403343 convert_dataset()
@ 0x402383 main
@ 0x7f409fd87f45 (unknown)
@ 0x4023c3 (unknown)
@ (nil) (unknown)
Aborted
修改:解压后的四个文件名字中的“.”变为“-”即可。如t10k-images.idx3-ubyte修改成t10k-images-idx3-ubyte。
问题得到解决
3.经过上述操作./examples/mnist/路径下会有mnist_test_lmdb和mnist_train_lmdb两个文件夹,分别是测试和训练数据。
在最终训练之前需要修改./examples/mnist/lenet_solver.prototxt最后一句话为,
solver_mode: CPU
这样保证整个训练过程在CPU上进行,最后在caffe的文件夹下执行训练命令,
./examples/mnist/train_lenet.st
就可以了