您的位置:首页 > Web前端

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

2017-07-01 21:04 483 查看
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


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐