zoukankan      html  css  js  c++  java
  • 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!

  • 相关阅读:
    Memcached使用与纠错(附代码和相关dll)
    python函数
    python操作文件
    python基础2
    SpringMVC中使用@ResponseBody注解将任意POJO对象返回值转换成json进行返回
    利用aspose-words 实现 java中word转pdf文件
    POI各Jar包的作用(转)
    java利用poi 把ppt转化为图片,
    SpringMVC中 解决@ResponseBody注解返回中文乱码
    springMVC 使用注解注入接口实现类
  • 原文地址:https://www.cnblogs.com/peony-jing/p/8617532.html
Copyright © 2011-2022 走看看