700字范文,内容丰富有趣,生活中的好帮手!
700字范文 > caffe-windows快速配置和测试训练教程

caffe-windows快速配置和测试训练教程

时间:2022-03-09 19:34:56

相关推荐

caffe-windows快速配置和测试训练教程

下载资源和数据

1、下载微软官方caffe-windows解压到F盘根目录。这个版本好处待会会说。点我进入Git下载

2、Cuda SDK。这里以Cuda7.5为例。解压到caffe-windows目录。点我进入官网下载

3、下载cuDNN。同样用7.5版本。点我进入官网下载。官网下载可能需要你注册,忍一下吧。

4、下载用于训练手写数字识别模型LeNet的mnist数据集。点我官网下载

编译caffe-windows

1、解压caffe-windows后,找到windows\CommonSettings.props.example文件,重命名为CommonSettings.props文件。然后重点找到下面几个属性进行修改。

<CpuOnlyBuild>false</CpuOnlyBuild><UseCuDNN>true</UseCuDNN><CudaVersion>7.5</CudaVersion><CudaArchitecture>compute_30,sm_30</CudaArchitecture><CuDnnPath>F:\caffe-master</CuDnnPath>

说明如下:

- CpuOnlyBuild: 指明只使用CPU。我们这里使用GPU,因此输入false。

- UseCuDNN: 指明使用GPU。因此输入true。显然和CpuOnlyBuild必然不能同真。

- CudaVersion: 你的cuda版本号!

- CudaArchitecture: 对应你的GPU的Cuda计算能力!请使用GPU-Z工具查看,鼠标放到CUDA单选框上即可。如下图所示。我这里是3.0.

- CuDnnPath: 指明cuDNN的存放目录。注意是cuda目录的父目录

2、使用Visual Studio 打开.\windows\Caffe.sln,编译之前请一定确认以下几个事项:

- 编译选项设置为Release,X64;

- ibcaffe工程属性→配置属性→C/C++→常规→将警告视为错误修改为“否”;

- libcaffe工程属性 →配置属性→ CUDA C/C++→Device →Code Generation中的CudaArchitecture确实设置正确(符合你的GPU计算能力);

3、开始编译解决方案。如果出现如下错误:

E:\NugetPackages\boost.1.59.0.0\lib\native\include\boost/format/alt_sstream_impl.hpp : error C2220: 警告被视为错误 - 没有生成“object”文件 (….\src\caffe\util\math_functions.cpp)

不要慌,编码方式问题而已。打开该文件,保存一下即可 1。

4、微软官方版本的好处是微软使用NuGet管理第三方开发包,编译过程会自动帮你下载一大堆依赖库到../NugetPackages目录中。因此可能会比较慢,但是毕竟方便。

5、全部生成成功后,caffe.exe等文件存放在.\Build\x64\Release目录中。

训练第一个手写数字识别网络——LeNet!

1、解压数据。

把下载好的mnist数据集解压到.\examples\mnist\mnist_data目录,可以看到其解压后由4个ubyte文件组成。

2、转换数据。

caffe必须将其转换成lmdb或者leveldb文件才能进行训练。在.\examples\mnist下新建一个Convert_Mnist_Data.bat文件,输入:

..\..\Build\x64\Release\convert_mnist_data.exe --backend=lmdb mnist_data\train-images.idx3-ubyte mnist_data\train-labels.idx1-ubyte mnist_train_lmdb..\..\Build\x64\Release\convert_mnist_data.exe --backend=lmdb mnist_data\t10k-images.idx3-ubyte mnist_data\t10k-labels.idx1-ubyte mnist_test_lmdbpause

运行成功后,就把mnist的训练数据和测试数据转换成了lmdb文件。注意其存放于.\examples\mnist下的mnist_test_lmdb和mnist_train_lmdb目录中。

3、计算训练数据均值。

在.\examples\mnist下新建一个Create_Mean.bat文件,输入:

..\..\Build\x64\Release\compute_image_mean.exe mnist_train_lmdb mean.binaryproto --backend=lmdbPause

运行成功后,当前目录会生成一个mean.binaryproto均值文件。训练中会用到。

4、修改LeNet网络描述文件

打开.\examples\mnist目录下的lenet_train_test.prototxt文件,查找transform_param属性,在其中加入一行代码,指定均值文件:

transform_param {mean_file: "mean.binaryproto"scale: 0.00390625}

打开.\examples\mnist目录下的lenet_solver.prototxt文件,注意最后几行:

snapshot_prefix: "examples/mnist/lenet"# solver mode: CPU or GPUsolver_mode: GPU

snapshot_prefix: 目录前缀,由于我们接下来会在examples\mnist目录下写一个bat来训练LeNet,因此,这里把目录修改为”lenet”即可。solver_mode: 指定使用GPU进行训练。你可以改为CPU来体验龟速训练。

5、开始训练

在.\examples\mnist下新建一个Train_Mnist.bat文件,输入:

..\..\Build\x64\Release\caffe.exe train --solver=lenet_solver.prototxtpause

使用GPU的话大概不到1分钟即可训练完毕,最后收敛时准确率在0.991左右。

测试自己的手写数字!

是不是迫不及待的想自己写几个数字让LeNet来识别了呢?部分博主提供了使用matlab转灰度图的方法2。这里我提供更加简单粗暴的方法如下:

1、使用画图程序,画布大小调整为较小的正方形即可;

2、使用墨水工具把背景涂成黑色;

3、铅笔手写一个数字,然后另存为单色位图即可,命名为0.bmp。如下图所示:

4、将图片存放到examples\mnist\mytest下。然后在该目录下新建一个标签描述文件label.txt,里面输入:

0 1 2 3 4 5 6 7 8 9

5、在.\examples\mnist下新建一个MyTest.bat文件,输入:

..\..\Build\x64\Release\classification.exe lenet.prototxt lenet_iter_10000.caffemodel mean.binaryproto mytest\label.txt mytest\0.bmppause

应该可以看到如下结果:

最后感谢其他网友提供的Check fail错误解决方案3,以及其他对mnist数据集4和Cifar数据集的训练教程5。

参考文献

/sunshine_in_moon/article/details/51426073 ↩/zb1165048017/article/details/52217772 ↩/yewei11/article/details/70212344 ↩/yixuan-xu/p/5862657.html ↩/zl20pn/p/6218958.html ↩

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。