zoukankan      html  css  js  c++  java
  • 上手Caffe(二)

    @author:oneBite
    本文简述如何在windows环境下,运行caffe的“hello world”例程
    体会适用caffe的流程:转换输入数据格式>在solver.prototxt中配置训练参数,在train.prototxt中配置网络结构和输入输出、激活函数>调用caffe train and test

    "hello world"-运行识别手写数字的例程

    先简要概述Linux下的运行caffe helloworld例程过程
    编译caffe>下载相应的训练数据到指定目录>然后运行sh文件训练和测试即可。
    Shell脚本所做的主要工作,调用编译得到相应的bin文件:
    1 将输入数据转化到指定的数据格式-LMDB。文件夹中附有源码:vert_mnist_data.cpp;
    2 运行caffe命令(即caffe.bin,是编译得到的可执行文件),训练网络>测试数据。在命令行中运行caffe –help可以看到些许帮助。

    Windows
    而windows下,solution编译之后也获得了很多exe文件,运行例程时,需要我们自己调用exe文件执行跟linux上一样的逻辑。
    在vs2013中,可以看到编译生成的可执行文件的输出路径。
    在project的properties>general>Output Directory中
    其中的变量可以参看https://msdn.microsoft.com/en-us/library/c02as0cs.aspx。在本机上,目录是caffe-windowscaffe-windowsBuildx64Debug。
    本次例程需要使用convert_mnist_data.exe,caffe.exe

    查看命令帮助:
    E:work17caffe-windowscaffe-windowsBuildx64Debug>
    .convert_mnist_data.exe –help
    使用命令时,也可以参看linux端的sh文件中是如何调用相应的bin文件的,有一定的参考价值。Caffe官方并没有给出较为详细的reference手册。
    convert_mnist_data.exe用于将mnist上下载的数据集转化为lmdb数据格式。可以看一下convert_mnist_data的源代码,了解lmdb格式内容。
    命令使用格式 [flag] 输入数据 输入label文件 输出文件夹

    Windows process

    上述大致介绍了caffe运行的原理。注意调用命令时,cmd所处的目录为caffe的主目录,本机为E:work17caffe-windowscaffe-windows。

    1 调用convert_mnist_data。
    E:work17caffe-windowscaffe-windows>路径convert_mnist_data 解压之后的image文件 解压之后的label文件 文件输出目录
    运行一次之后,在指定的输出目录下就会出现data.mdb和lock.mdb文件

    2 调用caffe,训练网络权值。(还是要注意运行exe时,cmd所处的当前目录)
    目录caffe.exe train --solver=目录lenet_solver.prototxt
    #solver.prototxt#
    定义了训练时所需的参数
    内容摘要:
    #定义了描述训练网络结构的prototxt文件的位置
    net: "examples/mnist/lenet_train_test.prototxt"
    # solver mode: CPU or GPU,这里需要进行更改,因为上次编译的caffe,修改了config文件,编译得到的是不适用GPU的caffe.exe
    solver_mode: CPU

    #lenet_train_test.prototxt#
    在lenet_train_test中设置训练和测试的数据来源 backend,大概格式的意思
    data_param {
    source: "examples/mnist/mnist_train_lmdb"
    batch_size: 64
    backend: LMDB
    }

    3 调用caffe,测试网络。
    权值文件 caffemodel
    caffe test -model train_test.prototxt -weights xxxx.caffemodel -iteration 100

  • 相关阅读:
    Django重要组件(Auth模块)
    Django框架深入了解(总结整理)
    ORM(数据库对象关系映射)代码重写
    Django框架深入了解——DRF之序列化、反序列化
    让无线更自由 TOTOLINK EX750无线中继评测
    matlab常用知识
    点云数据
    运动恢复结构
    Ubuntu下简单的QT绘图程序
    英语词语解释
  • 原文地址:https://www.cnblogs.com/ceo1207/p/6492119.html
Copyright © 2011-2022 走看看