• windows下使用caffe测试mnist数据集


    在win10机子上装了caffe,感谢大神们的帖子,要入坑caffe-windows的朋友们看这里,还有这里,安装下来基本没什么问题。

    好了,本博文写一下使用caffe测试mnist数据集的步骤。

    1. 下载mnist数据集。

    不太看得懂get_mnist.ps1文件,并且运行无效,所以选择直接从mnist官网下载数据集。下载后解压,从解压后的文件夹提取出四个文件,放在caffe根目录下<caffe-root>datamnist下,例如E:caffe-windowsdatamnist,否则后面运行报错找不到文件。

    2. 创建lmdb文件。

    也有的人会用leveldb,这里选择使用lmdb存储。在目录E:caffe-windowsexamplesmnist下,运行create_mnist.ps1文件,博主发现闪一下就退出了,抓屏后发现是找不到文件convert_mnist_data.exe,用编辑器打开,发现路径错误,看下图,这是修改过的,使用build_win.exe编译caffe时,build文件夹会建在scripts目录下,找到convert_mnist_data.exe的路径,修改create_mnist.ps1文件。可以看到$DATA后的数据,正是1中提取出的文件。

     

    再次运行,在examplesmnist下生成两个lmdb文件夹。如图:

     

    3. 修改参数

    examplesmnist目录下lenet_solver.prototxt文件默认使用GPU训练,根据自己情况使用CPU或者GPU。并将文件中的路径改为绝对路径,lenet文件夹是我自己创建的,貌似不能自动创建?

     

    同样,examplesmnist目录下lenet_train_test.prototxt文件,修改这里,绝对路径:

     

    4. 开始训练。

    改完参数了,创建bat文件训练。这里也可以使用examplesmnist目录下提供的train_lenet.ps1,运行出错,还不如用bat文件简简单单两句话

    E:caffe-windowsscriptsuild	oolsReleasecaffe.exe  train  --solver=E:caffe-windowsexamplesmnistlenet_solver.prototxt
    pause

    运行结果如图:

     

    Ok,done!

  • 相关阅读:
    一直追求技术好吗?
    设计模式前言
    vba 行 列 单元格 统计
    JavaScript 仿LightBox内容显示效果
    Excel 比大小
    多态
    JavaScript 图片切换展示效果
    vba 单元格 一系例操作
    关于Flex的争论(富客户端、弱客户端)
    input文本框样式代码实例
  • 原文地址:https://www.cnblogs.com/peony-jing/p/8617532.html
Copyright © 2020-2023  润新知