[caffe(一)]使用caffe训练mnist数据集

1.数据集的下载与转换

1）我们在mnist数据集上做测试，MNIST handwritten digit database, Yann LeCun, Corinna Cortes and Chris Burge在这里下载这四个文件：

2）然后解压生成了以下四个文件：

3）数据格式转换

新建一个文本文档，更改后缀为 .bat 在里面复制以下code：

..\..\Build\x64\Release\convert_mnist_data.exe ..\..\data\mnist\train-images.idx3-ubyte ..\..\data\mnist\train-labels.idx1-ubyte ..\..\examples\mnist\mnist_train_lmdb

echo.

..\..\Build\x64\Release\convert_mnist_data.exe ..\..\data\mnist\t10k-images.idx3-ubyte ..\..\data\mnist\t10k-labels.idx1-ubyte ..\..\examples\mnist\mnist_test_lmdb

pause

其实第一个路径就是 exe文件所在路径，后面两个就是数据所在路径，最后一个是输出文件路径

例如：

F:\caffe-master\caffe-master\Build\x64\Release\convert_mnist_data.exe E:\caffetest\mnist\mnist\train-images.idx3-ubyte E:\caffetest\mnist\mnist\train-labels.idx1-ubyte E:\caffetest\mnist\mnist\mnist_train_lmdb

echo.

F:\caffe-master\caffe-master\Build\x64\Release\convert_mnist_data.exe E:\caffetest\mnist\mnist\t10k-images.idx3-ubyte E:\caffetest\mnist\mnist\t10k-labels.idx1-ubyte E:\caffetest\mnist\mnist\mnist_test_lmdb

pause

保存后，双击运行，生成了两个文件夹：mnist_train_lmdb和mnist_test_lmdb

2.打开lenet_train_test.prototxt

然后打开lenet_solver.prototxt,修改第二行即可，

可以将prototxt文件放在以下的网址，查看网络的结构：http://ethereon.github.io/netscope/#/editor

3.使用caffe.exe 训练网络

利用命令行cd到caffe.exe的工作目录，输入以下的命令：

caffe train -solver=lenet_solver.prototxt -gpu 0

可以得到以下的训练结果：

4.使用Python调用caffe

环境：python+Anaconda；同时将编译生成的python文件夹放在..\anaconda2\Lib\site-packages中。

import caffe
caffe.set_mode_gpu()
solver = caffe.SGDSolver('lenet_solver.prototxt')
solver.solve()

运行代码可以得到以下的结果：

转载于:https://www.cnblogs.com/youngsea/p/9430875.html

[caffe(一)]使用caffe训练mnist数据集相关推荐

实践详细篇-Windows下使用VS2015编译的Caffe训练mnist数据集
上一篇记录的是学习caffe前的环境准备以及如何创建好自己需要的caffe版本.这一篇记录的是如何使用编译好的caffe做训练mnist数据集,步骤编号延用上一篇 <实践详细篇-Windows下 ...
TensorFlow精进之路（十四）：RNN训练MNIST数据集
1.概述前面介绍了RNN,这一节就用tensorflow的RNN来训练MNIST数据集,看看准确率如何. 2.代码实现 2.1.导入数据集 # encoding:utf-8 import tenso ...
深度学习基础: BP神经网络训练MNIST数据集
BP 神经网络训练MNIST数据集不用任何深度学习框架,一起写一个神经网络训练MNIST数据集本文试图让您通过手写一个简单的demo来讨论 1. 导包 import numpy as np imp ...
使用caffe自带模型训练cifar10数据集
前面训练了mnist数据集!但caffe自带的数据集还有cifar10数据集.同样cifar10数据集也是分类数据集,共分10类.cifar10数据集中包含60000张32x32的彩色图片.(其中 ...
ubuntu16.04caffe训练mnist数据集
搭好了环境,下面就该训练模型了呀!实践才是真理的唯一标准!大多数情况下,新接触caffe的小白们第一个训练的模型一定是Mnist数据集吧.这篇文章就以mnist数据集为例介绍下如何训练模型吧!(训 ...
Pytorch 实现全连接神经网络/卷积神经网络训练MNIST数据集，并将训练好的模型在制作自己的手写图片数据集上测试
使用教程代码下载地址:点我下载模型在训练过程中会自动显示训练进度,如果您的pytorch是CPU版本的,代码会自动选择CPU训练,如果有cuda,则会选择GPU训练. 项目目录说明: CNN文件夹 ...
训练MNIST数据集模型
1. 数据集准备详细信息见: Caffe: LMDB 及其数据转换 mnist是一个手写数字库,由DL大牛Yan LeCun进行维护.mnist最初用于支票上的手写数字识别, 现在成了DL的入门练习 ...
Caffe概览及mnist数据集训练
1. Caffe目录结构 data/用于存放下载的训练数据 docs/ 帮助文档 examples/ 代码样例 matlab/ MATLAB接口文件 python/ PYTHON接口文件 models ...
(caffe入门)windows caffe 之 mnist 训练
文章目录 1. mnist 数据集下载 2. mnist 数据集转换 3. 修改网络结构文件 lenet_train_test.prototxt 和网络求解文件 lenet_solver.proto ...

[caffe(一)]使用caffe训练mnist数据集

[caffe(一)]使用caffe训练mnist数据集相关推荐

最新文章

热门文章