Caffe修炼之路(2)——运行MNIST示例


在这里插入图片描述

(一)前 言

在上一节中,我们成功安装了Caffe,接下来就先来探索一番,还是熟悉的数据集,还是熟悉的味道:
在这里插入图片描述
这一节我们来跑一跑Caffe自带的MNIST示例,使用Clion打开Caffe-master文件夹,如下所示:
在这里插入图片描述我们要用的都在examples/mnist目录下

(二)下载MNIST数据

先来下载MNIST数据,键入下列命令:

cd caffe-master/data/mnist
bash get_mnist.bash

输出:
在这里插入图片描述

(三)MNIST数据格式转换

在caffe-master目录下执行:

bash examples/mnist/create_mnist.sh

在这里插入图片描述

(四)运行示例

因为我们装的CPU版Caffe,所以需要改一下prototxt文件,在examples/mnist下找到lenet_solver.prototxt文件,将solver_mode设定为CPU:

solver_mode: CPU

在caffe-master目录下执行:

 build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

开始训练:
在这里插入图片描述等着模型跑完就ok了。

发布了80 篇原创文章 · 获赞 74 · 访问量 32万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 黑客帝国 设计师: 上身试试

分享到微信朋友圈

×

扫一扫,手机浏览