caffe+win10+mnist数据集(CPU only)

  1. 获取mnist数据
    官网下载:http://yann.lecun.com/exdb/mnist/
    解压后存入.\data\mnist (. 是caffe根目录)
    caffe+win10+mnist数据集(CPU only)_第1张图片
  2. 转换成lmdb格式数据
    在.\examples\mnist\下建立文件create_mnist.bat,内容如下:
echo "create train data"
.\scripts\build\examples\mnist\Release\convert_mnist_data.exe .\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb
echo "create test data" 
.\scripts\build\examples\mnist\Release\convert_mnist_data.exe .\data\mnist\t10k-images.idx3-ubyte .\data\mnist\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb
pause

第一个是convert_mnist_data.exe所在路径,第二个是第一步中获取到数据的路径,最后一个是输出数据。输出的mnist_train_lmdb和mnist_test_lmdb文件夹中各有data.mdb和lock.mdb两个文件。
保存后双击运行。
caffe+win10+mnist数据集(CPU only)_第2张图片

  1. 修改.\examples\mnist\lenet_solver.prototxt,最后一行
solver_mode:CPU
  1. caffe根目录建立文件train_mnist.bat,内容如下:
.\scripts\build\tools\Release\caffe.exe train --solver=.\examples\mnist\lenet_solver.prototxt 
pause

保存后双击运行。
caffe+win10+mnist数据集(CPU only)_第3张图片

你可能感兴趣的:(深度学习)