windows环境Caffe训练mnist

    xiaoxiao2021-12-14  20

    (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   .\Build\x64\Debug\caffe.exetrain --solver=.\examples\mnist\lenet_solver.prototxt   pause  

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

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

    转载请注明原文地址: https://ju.6miu.com/read-970598.html

    最新回复(0)