用caffe在windows平台下训练mnist数据集

前面博客已经提到了怎么在windows上安装和编译caffe,接下来是在windows上用CPU跑mnist数据集

1 将caffe的可执行文件加入环境变量

上一步编译完之后,在debug目录下生成了caffe的exe文件,将这些可执行文件加入环境变量就可以在系统的任意位置访问它
用caffe在windows平台下训练mnist数据集

2 下载mnist数据集

mnist数据集下载地址 : http://yann.lecun.com/exdb/mnist/

3 修改配置文件

需要修改文件 lenet_solver.prototxtlenet_train_test.prototxt
lenet_solver.prototxt 中最后以后的GPU改为CPU(或者按你的情况来)
lenet_train_test.prototxt 中按你的实际路径修改两个source的值
用caffe在windows平台下训练mnist数据集

用caffe在windows平台下训练mnist数据集

运行模型

在我的电脑上运行了不到80分钟左右(电脑比较渣)。结果如下:
用caffe在windows平台下训练mnist数据集