caffe中 cifar10案例(一)训练模型

1. 背景知识
cifar10数据训练样本50000张,测试样本10000张,每张为32*32的彩色三通道图片,共分为10类。

2. 下载数据:

$ cd /opt/caffe-master

$ sudo sh data/cifar10/get_cifar10.sh

直接运行这个脚本速度特慢,用迅雷直接下载http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz,放到 /opt/caffe-master/data/cifar10下,并解压,出来一大堆bin文件。

3. 转换数据格式为lmdb

$ sh examples/cifar10/create_cifar10.sh

转换成功后,会在 examples/cifar10/文件夹下生成两个文件夹,cifar10_train_lmdb和cifar10_test_lmdb,里面的文件就是我们需要的文件。
由于没有配置GPU,因此在examples/cifar10/cifar10_quick_solver.prototxt和 cifar10_quick_solver_lr1.prototxt文件中,最后一行,设为solver_mode: CPU
为了节省时间,进行快速训练(train_quick),训练分为两个阶段,第一个阶段(迭代4000次)调用配置文件cifar10_quick_solver.prototxt,学习率(base_lr)为0.001。

4. 训练数据

 $sudo time sh examples/cifar10/train_quick.sh

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值