【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
Greatness is never a given, it must be earned.