zoukankan      html  css  js  c++  java
  • ubuntu16.04+caffe训练mnist数据集

    1.   caffe-master文件夹权限修改


    下载的caffe源码编译的caffe-master文件夹貌似没有写入权限,输入以下命令修改:

    sudo chmod -R 777 ~/caffe-master/


    2.   下载mnist数据库


    cd ~/caffe
    sduo ./data/mnist/get_mnist.sh


    caffe中的./data/mnist/get_mnist.sh 文件实现了下载mnist数据库的功能,文件的内容如下:



    执行之后,在./data/mnist文件夹下生成4个文件,分别是测试、训练数据库和测试、训练标签: 




    3.   二进制数据库文件转换成lmdb数据库格式 


    sudo ./examples/mnist/create_mnist.sh

    create_mnist.sh文件用于把数据库转换成lmdb格式,内容如下:



    执行之后,在 ./examples/mnist文件夹下生成两个文件夹,一个是 mnist_train_lmdb 和 mnist_test_lmdb,分别存放了训练和测试数据:




    4.   训练lenet网络


    如果只是用CPU训练的话,需要先在 lenet_solver.prototxt 文件中修改训练模式为CPU,使用gedit打开.prototxt文件:

    sudo gedit ./examples/mnist/lenet_solver.prototxt


    修改文件最后一行的 solver_mode的配置为 CPU。


    以下指令执行训练:

    sudo ./examples/mnist/train_lenet.sh


    训练的最终结果:



    训练结果的分类文件 lenet_iter_10000.caffemodel 文件保存在 ./examples/mnist路径下。


  • 相关阅读:
    vue计算属性和方法的区别
    函数防抖和函数节流
    vue项目使用keep-alive
    hash模式与history模式
    Vue中的计算属性
    MVVM的理解和Vue的生命周期
    session和cookie的区别
    localStorage和sessionStorage区别
    try catch finally的理解
    《Linux命令学习手册》系列分享专栏
  • 原文地址:https://www.cnblogs.com/mtcnn/p/9411831.html
Copyright © 2011-2022 走看看