caffe---mnist数据集训练与测试

最新推荐文章于 2025-08-07 15:24:42 发布

weixin_30757793

最新推荐文章于 2025-08-07 15:24:42 发布

阅读量72

点赞数

CC 4.0 BY-SA版权

文章标签：人工智能

原文链接：http://www.cnblogs.com/crazybird123/p/7788746.html

本文详细介绍了如何使用Caffe深度学习框架搭建MNIST手写数字识别系统，包括配置训练和测试数据路径、生成模型快照、计算均值文件及进行图像分类测试等步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、数据。mnist_test_lmdb和mnist_train_lmdb数据

2、路径。

（1）修改lenet_train_test.prototxt文件，训练和测试两处

  source: "....省略/examples/mnist/mnist-train-leveldb" //写上你的绝对路径  backend: LEVELDB //格式改成LEVELDB

（2）修改lenet_solver.prototxt文件：

net: "....省略/examples/mnist/lenet_train_test.prototxt"  //绝对路径

snapshot_prefix: "....省略/examples/mnist/lenet" //绝对路径

solver_mode: CPU //CPU模式

3、右键caffe打开属性：

在Command Arguments输入： train --solver=前面的绝对路径/mnist/lenet_solver.prototxt

4、确定后debug caffe,大功告成！

1、训练完后，会生成lenet_iter_5000.caffemodel，lenet_iter_5000.solverstate，lenet_iter_10000.caffemodel，lenet_iter_10000.solverstate四个文件

2、产生均值文件

计算均值文件：在E:\CaffeDev-GPU\caffe-master\Build\x64\Release目录下新建bat文件mnist_mean.bat，内容如下：

D:/caffewin/caffe-master/Build/x64/Debug/compute_image_mean.exe D:/caffewin/caffe-master/examples/mnist/mnist_train_lmdb mean.binaryproto --backend=lmdb
pause

双击运行

得到mean.binaryproto

3、新建mnist_test.bat

D:/caffewin/caffe-master/Build/x64/Debug/caffe.exe test --model=D:/caffewin/caffe-master/examples/mnist/lenet_train_test.prototxt --weights=D:/caffewin/caffe-master/examples/mnist/lenet_iter_10000.caffemodel
pause

双击运行

4、测试图片

新建test.bat文件，写入

D:\caffewin\caffe-master\Build\x64\Debug\classification.exe D:\caffewin\caffe-master\examples\mnist\lenet.prototxt D:\caffewin\caffe-master\examples\mnist\lenet_iter_10000.caffemodel D:\caffewin\caffe-master\examples\mnist\mean.binaryproto D:\caffewin\caffe-master\examples\mnist\words.txt D:\caffewin\caffe-master\examples\mnist\2.bmp  
pause

双击运行即可

转载于:https://www.cnblogs.com/crazybird123/p/7788746.html