windows环境Caffe训练mnist

来源:互联网 发布:sql server 混合模式 编辑:程序博客网 时间:2024/06/06 14:09

(1)去官网http://yann.lecun.com/exdb/mnist/下载mnist数据集。


解压放到D:\caffe-master\data\mnist目录下;

(2)在caffe根目录下,新建一个mycreate_mnist.bat,里面写入如下的脚本。

.\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb
echo.   
.\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\t10k-images.idx3-ubyte   .\data\mnist\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb
pause  

然后双击该脚本运行,即可在E:\caffe\examples\mnist下面生成两个文件夹,及相应的lmdb数据文件。


注意:如果该文件夹存在或文件夹里的文件存在,会出现错误,需要删除后再生成


(3)修改E:\caffe\examples\mnist\lenet_solver.prototxt,将最后一行改为solver_mode:CPU

修改E:\caffe\examples\mnist\lenet_train_test.prototxt,讲train和test数据地址修改成lmdb文件夹

data_param {
    source: "D:/caffe-master/examples/mnist/mnist_train_lmdb"
    batch_size: 64
    backend: LMDB
  }

data_param {
    source: "D:/caffe-master/examples/mnist/mnist_test_lmdb"
    batch_size: 100
    backend: LMDB
  }

(4)在caffe根目录下,cmd中运行如下命名

[plain] view plain copy
 在CODE上查看代码片派生到我的代码片
  1. .\Build\x64\Debug\caffe.exetrain --solver=.\examples\mnist\lenet_solver.prototxt  
  2. pause  

就会开始训练,训练完毕后会得到相应的准确率和损失率。


i3破电脑上运行的时间是:11分钟



0 0
原创粉丝点击