完整教程:使用caffe测试mnist数据集

这篇原创笔记来自铁粉zhupc,感谢为大家提供的这份caffe测试mnist数据集的精彩总结。

想要入门深度学习没有几个趁手的兵器是不行的,目前流行的框架tensorflow、pytorch、caffe等,笔者也是最近接触了caffe,发现caffe非常适合初学者入门深度学习。不必像tensorflow那样,先学习Python,然后在学习tf,这个过程感觉像在重新学习一门语言。caffe是c++编写的,所以从github上下载下来后需要你自己编译,令人高兴的是caffe也支持windows,你去github上下载微软发布的caffe用vs2013编译即可成功,它也区分cpu版本与gpu版本,如何编译安装的百度上教程基本可用,笔者在windows跟ubuntu都编译成功了。入门的童鞋基本上接触的第一个教程就是mnist识别手写数字,caffe无疑可以让你最快的搭建整个网络并跑通。

这里假设你已经安装好caffe。caffe给我们配备了很多个例子,安装好的caffe中,在example文件夹下,有很多现成的网络,以mnist为例。首先,我们需要下mnist数据集,在进入到data文件夹下,有个获取数据的脚本 caffe/data/mnist/get_mnist.sh,执行完成后会得到下面几个文件,通过名字判断可知道分别是测试集与训练集的样本与标签。

进入到caffe/examples/mnist 文件夹下,执行./create_mnist.sh 脚本,如果你的caffe没有编译,可能会提示错误convert_mnist_data.bin: not found。 重新编译一下caffe即可。脚本执行成功后你会得到两个文件夹

Lmdb是一种数据库,查询和插入非常高效,caffe使用lmdb作为数据源,同时caffe也支持hdf5文件。

Caffe搭建网络是基于prototxt文件,超参数也在里面配置。所以只需要根据自己的需求配置网络与超参数prototxt文件,就可以了。在mnist目录下,有以下几个文件:

找到 文件caffe/examples/mnist/lenet_train_test.prototxt 这个文件是训练用的文件,想要自定义网络就可以这个文件配置网络。

简单粗暴的配置,相信你很容易看懂,配置数据源,定义layer类型。

你还需要caffe/examples/mnist/lenet_sover.prototxt文件。lenet_solver文件是用来配置超参数,打开这个文件

框选出来的是几个重要的配置,首先定义网络文件位置,配置测试集迭代次数,定义学习率。以及最大迭代次数,文件末尾也可以自由的定义使用GPU或者CPU,snapshot_prefix指的是快照生成的路径,这里要配置好。另外建议,数据源最好配置上绝对路径,这样的话在任何目录下都可以执行下面命令。

找到tools文件夹下的caffe,配置solver路径后,即可运行。

有的童鞋可能用我的命令执行不通过,你只需要查看三个路径是否配置正确,一个是solver文件中的 net 路径,跟快照路径,网络文件中的数据源路径。还需要注意的是你在什么路径下执行 train命令。

原文发布于微信公众号 - Python与机器学习算法频道(alg-channel)

原文发表时间:2018-11-07

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏木头编程 - moTzxx

小程序 POST请求不到数据的一种情况

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u011415782/article/de...

2213
来自专栏hotqin888的专栏

ONLYOFFICE权限开发之一

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/hotqin888/article/det...

2374
来自专栏hotqin888的专栏

HydromCMS增加封面、完善自定义目录功能、目录结构的修改功能

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/hotqin888/article/det...

1142
来自专栏Petrichor的专栏

Tensorflow源码 目录树

很多博客中提到的 models/(位于根目录下,该目录下存放这多个使用python实现的模型实例),我并没有看到。估计是最新版的Tensorflow源码取消了这...

2761
来自专栏CreateAMind

ls-gan bedroom数据集 pretrain model 和代码

pretrain model 和代码 https://pan.baidu.com/s/1dFbzjlZ pretrain 下载

1082
来自专栏debugeeker的专栏

《coredump问题原理探究》windows版第四章函数的逆向

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/xuzhina/article/detai...

941
来自专栏Python入门

python爬虫使用scrapy框架爬取顶点小说网

scrapy框架 是一个非常好的东西,能够实现异步爬取,节省时间,其实本文纯粹的按照之前的思维来做,

1860
来自专栏ml

PIL中的Image和numpy中的数组array相互转换

1. PIL image转换成array img = np.asarray(image) 需要注意的是,如果出现read-only错误,并不是转换的错...

3959
来自专栏技术博文

smarty模板引擎原理解析

//php 控制器文件 <?php //引入模板引擎文件 include("20130304.php"); $smarty = new TinySmarty()...

3406
来自专栏互联网杂技

利用机器学习把草图自动生成HTML

项目地址 https://github.com/ashnkumar/sketch-code 可以先clone到本地

1582

扫码关注云+社区

领取腾讯云代金券