Windows平台基于Caffe框架的LeNet网络训练

在Windows平台下使用Caffe的确不如Linux来的方便，至少人家把Shell都已经写好了。但是像我这种VS重度依赖者，还是离不开微软大腿呀…废话不多说，一步步来吧

0. 为了后续文件路径访问的便利，我们先将$CAFFE_ROOT根目录添加到操作系统环境变量PATH中，并重启使之生效。

1. 首先需要从Yann LeCun的网站 http://yann.lecun.com/exdb/mnist/  上下载手写数字图像库，共为四个压缩包约11MB大小，分别是train-images-idx3-ubyte.gz（训练集图像），train-labels-idx1-ubyte.gz（训练集标签），t10k-images-idx3-ubyte.gz（测试集图像）和t10k-labels-idx1-ubyte.gz（测试集标签）。并将其解压缩至Caffe目录下的$CAFFE_ROOT\data\mnist文件夹中。

2. 然后我们需要将解压缩后的四个文件转为Caffe所支持的lmdb文件，转换需要使用到编译好的convert_mnist_data.exe可执行文件。由于Windows不能执行Caffe提供的Shell文件，在此将其修改为bat批处理文件内容如下：

@echo off

rd /s /q mnist_test_lmdb
rd /s /q mnist_train_lmdb

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

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

pause

将bat文件放在$CAFFE_ROOT\examples\mnist目录下双击运行，转换成功如下图所示：

3. 这时准备工作就已经就绪，我们可以开始对数据集进行训练了。

@echo off

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

pause

或者我们可以在Visual Studio中更改工程属性，配置调试命令参数，

执行批处理文件或直接在Visual Studio中执行调试，Caffe就运行起来了。在NVIDIA Geforce GTX970机器上，LeNet迭代10000次大约只需要1分钟就训练好了（在Ubuntu系统上训练时间更短，只需要30s左右，原因是Linux版cuDNN效率更高），结果如下图所示