Caffe 快速上手MNIST数据集分类(windows)

  1. 下载数据集
    http://yann.lecun.com/exdb/mnist/
    在这里插入图片描述
    为了简便我直接下载到你的path\caffe-master\examples\mnist,并创建文件夹MNIST_data,将数据解压后存放于此处,如图所示:
    在这里插入图片描述

  2. 下载的原始数据集为二进制文件,需要转换为LEVELDB或LMDB格式;
    存放位置:
    在这里插入图片描述
    convert_train_lmdb.bat:

    D:\Caffe\caffe-master\Build\x64\Release\convert_mnist_data.exe ^
    D:\Caffe\caffe-master\examples\mnist\MNIST_data\train-images.idx3-ubyte ^
    D:\Caffe\caffe-master\examples\mnist\MNIST_data\train-labels.idx1-ubyte ^
    D:\Caffe\caffe-master\examples\mnist\lmdb\train_lmdb
    pause
    

    convert_test_lmdb.bat:

    D:\Caffe\caffe-master\Build\x64\Release\convert_mnist_data.exe ^
    D:\Caffe\caffe-master\examples\mnist\MNIST_data\t10k-images.idx3-ubyte ^
    D:\Caffe\caffe-master\examples\mnist\MNIST_data\t10k-labels.idx1-ubyte ^
    D:\Caffe\caffe-master\examples\mnist\lmdb\test_lmdb
    pause
    

    运行结束生成文件:
    在这里插入图片描述

  3. 修改网络模型描述文件
    在这里插入图片描述
    lenet_train_test.prototxt:
    在这里插入图片描述
    在这里插入图片描述

  4. 修改超参数文件
    在这里插入图片描述
    lenet_solver.prototxt:
    在这里插入图片描述

  5. 开始训练模型
    在这里插入图片描述
    train.bat:

    D:\Caffe\caffe-master\Build\x64\Release\caffe.exe train ^
    -solver=D:/Caffe/caffe-master/examples/mnist/lenet_solver.prototxt
    pause
    
  6. 等待模型训练好,准备要测试的图片
    由于我使用的GPU进行训练,所以不到一分钟大概就可以训练完成,得到如下结果:
    在这里插入图片描述
    中间需要很多次迭代…
    在这里插入图片描述
    在这里插入图片描述
    准备的测试图片:
    在这里插入图片描述

  7. 生成均值文件
    在这里插入图片描述
    mean_file.bat:

    %计算均值程序%
    %train_lmdb%
    %均值文件存放路径%
    
    D:\Caffe\caffe-master\Build\x64\Release\compute_image_mean.exe ^
    D:\Caffe\caffe-master\examples\mnist\lmdb\train_lmdb ^
    D:\Caffe\caffe-master\examples\mnist\mean_file\mean.binaryproto
    pause
    
  8. 准备标签
    在这里插入图片描述
    label.txt:
    在这里插入图片描述

  9. 测试分类效果
    在这里插入图片描述
    mnist_classification.bat:

    %分类可执行程序%
    %网络结构%
    %训练好的模型%
    %均值文件%
    %标签%
    %要分类的图片%
    
    D:\Caffe\caffe-master\Build\x64\Release\classification.exe ^
    D:\Caffe\caffe-master\examples\mnist\lenet.prototxt ^
    D:\Caffe\caffe-master\examples\mnist\models\models_iter_10000.caffemodel ^
    D:\Caffe\caffe-master\examples\mnist\mean_file\mean.binaryproto ^
    D:\Caffe\caffe-master\examples\mnist\label\label.txt ^
    D:\Caffe\caffe-master\examples\mnist\MNIST_data\0-9\5.bmp
    pause
    

    运行批处理文件之后的结果:
    在这里插入图片描述
    从以上结果可以知道,最有可能的结果是5。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值