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

@tags: caffe python

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

那就写个python脚本替代吧。

step1

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

caffe_build D:libcaffe-masterBuildx64Release
caffe_root D:libcaffe-master

step2

在<caffe_root>examplesmnist下新建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 examplesmnist	rain_lenet.py