微软caffe+mnist训练及预测

    xiaoxiao2021-03-25  137

                                                   训练mnist数据集

    第一步:去官网下载mnist数据集http://yann.lecun.com/exdb/mnist/

    下载解压后放入caffe-master\data\mnist

    第二步:在caffe-master根目录下,创建create-mnist.bat,写入如下脚本

    .\Build\x64\Release\convert_mnist_data.exe.\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte.\examples\mnist\mnist_train_lmdb  

    echo.  

    .\Build\x64\Release\convert_mnist_data.exe.\data\mnist\t10k-images.idx3-ubyte  .\data\mnist\t10k-labels.idx1-ubyte.\examples\mnist\mnist_test_lmdb 

    Pause

    保存后双击该脚本运行就可以在caffe\examples\mnist下生成相应的lmbd数据文件

    第三步:在caffe-master\examples\mnist\lenet_solver.prototxt,将最后一行改为solver_mode:CPU

    不修改就会报错:check failure stack trace

    修改caffe-master\examples\mnist\lenet_train_test.prototxt中的source 把文件位置对应起来

    第四步:生成均值文件mean.binaryproto

    产生均值文件的方法是利用解决方案中的compute_image_mean.exe,位于目录\caffe-windows\Build\x64\Release下。回到caffe-windows根目录下创建一个mnist_mean.txt,写入如下内容:

    Build\x64\Release\compute_image_mean.exeexamples\mnist\mnist_train_lmdb mean.binaryproto --backend=lmdb

    pause

       将后缀名改为bat后双击运行。正确运行的话会在根目录下产生一个mean.binaryproto,也就是我们所需要的均值文件。接着为了使用均值文件需要稍微修改下层的定义。所以打开\examples\mnist\lenet_train_test.prototxt,做如下修改:

     

    第五步:在caffe-master根目录下 新建train_mnist.bat,输入如下脚本

    Build\x64\Release\caffe.exe  train--solver=examples/mnist/lenet_solver.prototxt 

    pause 

    保存双击运行,开始训练

    或者在cmd里面到caffe-master目录下  输入Build\x64\Release\caffe.exe  train--solver=examples/mnist/lenet_solver.prototxt,同样可以训练

    利用HDF5文件训练MNIst

    首先、利用matlabmnsit转换为HDF5文件,matlab程序如下

    clc

    close all

    clear all

    %%

     

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

    最新回复(0)