准备数据集
在终端上运行以下指令:
cd caffe/data/cifar10
./get_cifar10.sh
cd caffe/examples/cifar10
./create_cifar10.sh
其中caffe是你的根目录。
此时会出现报错:
./build/examples/cifar10/convert_cifar_data.bin: not found
但素! convert_cifar_data.bin
明明就在 build/examples/cifar10 中啊!
1.先用文本编辑器打开 create_cifar10.sh 看看里面有些啥,发现其中 convert_cifar_data.bin 路径不一致。
只需将
/examples/cifar10/convert_cifar_data.bin: not found
修改为
/build/examples/cifar10/convert_cifar_data.bin: not found
2.在根目录运行
注意!
新版caffe必须在根目录下执行
cd caffe
./examples/cifar10/create_cifar10.sh
运行之后,在 examples/cifar10 中会出现 cifar10_test_leveldb和cifar10_train_lmdb 两个文件
训练与测试
注意!
同样也需要在根目录中运行
cd caffe
./examples/cifar10/train_quick.sh
运行成功!
当5000次迭代之后,正确率约为75.2%
参考信息:http://caffe.berkeleyvision.org/gathered/examples/mnist.html