windows运行caffe实例

【参考1】http://blog.csdn.net/u012878523/article/details/41698209
【参考2】http://blog.csdn.net/u013657981/article/details/49497753
【总结下在windows上用caffe跑自己数据流程【
我是在建立在caffe-windows-master中建立data文件夹存放bat文件,数据集和生成的“leveldb”“mean.binaryproto””train.txt””val.txt”存放在data中的mine文件夹中

1、 数据预处理
(1)生成list,将数据集划分为train,test文件夹,运行批处理文件,将F:\Image Dataset\Corel\
换成图像集存放的目录
@dir/b/s/a-d "F:\Image Dataset\Corel\*.*">"F:\Image Dataset\Corel\list.txt”
(2)给目录打标签:写了一个matlab程序,写的太不美观,就不分享了,如果数据比较少,可以手动打标签。
生成的list的格式为“Butterfly/52099.JPG 0”,【注意】标签从0开始;用的是“/”而不是“\”

2、 生成leveldb和lmdb格式的文件,bat文件参数表示 重定义图像尺寸256*256 shuffle乱序 生成“leveldb”格式 数据集所在的路径 list目录所在的路径 生成的“leveldb”所在的路径

GLOG_logtostderr=1
%echo ["create train leveldb"]
convert_imageset.exe   
 --resize_height=256 --resize_width=256
 --shuffle 
--backend="leveldb" 
F:/caffe-windows-master/data/mine/train/ F:/caffe-windows-master/data/mine/train_list.txt  F:/caffe-windows-master/data/train_leveldb 0
%echo ["create val leveldb"]
convert_imageset.exe --resize_height=256 --resize_width=256 --shuffle --backend="leveldb" F:/caffe-windows-master/data/mine/val/ F:/caffe-windows-master/data/mine/test_list.txt  F:/caffe-windows-master/data/val_leveldb 0
echo [done]
pause

【注意】这里默认是生成“lmdb”格式,但是运行bat,会报错“check failed = mdb_status” == 0<112 vs 0>磁盘空间不足
路径的“/”和“\”要注意区分,用到list的路径都用“/”,因为是linux下的路径,=
3、 生成mean文件,mean空间基本上是一张白色的图像,用于样本和mean做差值输入。

compute_image_mean.exe --backend="leveldb" F:\caffe-windows-master\data\mine\train_leveldb F:\caffe-windows-master\data\mine\image_mean.binaryproto

4、 万事具备,只欠训练了,运行run_mine.bat

caffe.exe train --solver=mine/lenet_solver.prototxt
solver_mode: GPU(如果没有GPU,改成CPU)
 covering the full 10,000 testing images.
test_iter: 10
Carry out testing every 500 training iterations.
test_interval: 20
test_iter*test_iter = 200 一次测试的图片数,大致等于test的样本总数。
Mnist_demo_LeNet.prototxt 调用的网络的格式,
更改 source: "mine/train_leveldb",为leveldb存放目录
source: "mine/val_leveldb"
更改 iP2层的numout = 2 (自己数据的类别数)

5、运行生成lenet_iter_***.caffemodel

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值