博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
ubuntu16.04+caffe训练mnist数据集
阅读量:5031 次
发布时间:2019-06-12

本文共 798 字,大约阅读时间需要 2 分钟。

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

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

sudo chmod -R 777 ~/caffe-master/

2.   下载mnist数据库

cd ~/caffesduo ./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路径下。

转载于:https://www.cnblogs.com/mtcnn/p/9411831.html

你可能感兴趣的文章
发布高性能迷你React框架anu
查看>>
Python中Gradient Boosting Machine(GBM)调参方法详解
查看>>
利用DDE通信将PLC数据传输到EXCEL
查看>>
Eclipse 实用快捷键大全
查看>>
与非门和或门实现异或门
查看>>
golang统计出其中英文字母、空格、数字和其它字符的个数
查看>>
poj 1782 Run Length Encoding
查看>>
《自我介绍》
查看>>
在线考试系统设计思路
查看>>
p1150[noip2013普及]表达式求值
查看>>
POST和GET有什么区别?
查看>>
js基础
查看>>
mmsplayer for ios v1.0
查看>>
Python ghost.py 0.2版简单实例
查看>>
Python3安装cx_Oracle连接oracle数据库实操总结
查看>>
error:crosses initialization of ...
查看>>
SQL Server TRY...CATCH
查看>>
提供推荐
查看>>
leetcode 72. Edit Distance
查看>>
字符串移位包含的问题
查看>>