【caffe】train_lenet.sh在windows下的解决方案

@tags: caffe python

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

那就写个python脚本替代吧。

step1

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

caffe_build D:\lib\caffe-master\Build\x64\Release
caffe_root D:\lib\caffe-master

step2

在<caffe_root>\examples\mnist\下新建train_lenet.py:

# train_lenet.py

import os

caffe_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

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

python examples\mnist\train_lenet.py