@tags: caffe python

在windows下配置caffe后，跑mnist手写数字识别的例子。发现train_lenet.sh不能运行。

那就写个python脚本替代吧。

step1

定义两个环境变量，请自行添加并替换成你的目录：

caffe_build D:\lib\caffe-master\Build\x64\Releasecaffe_root D:\lib\caffe-master

step2

在\examples\mnist\下新建train_lenet.py:

# train_lenet.pyimport oscaffe_root=os.environ['caffe_root']caffe_build=os.environ['caffe_build']cmd=caffe_build+"\\caffe.exe train --solver="+caffe_root+"\\examples\\mnist\\lenet_solver.prototxt"os.system(cmd)

step3

在目录下执行:(一定要在这个目录下）

python examples\mnist\train_lenet.py