Caffe 学习笔记之CIFFA-10
背景知识
今天小编就亲身实践利用前几天在Ubuntu14.04刚装好的caffe进行CIFFA-10的训练。
CIFAR-10 数据集包含 60000张 32x32的彩色图片,一共有十种类别,每种类别有6000张。数据集中有50000张训练集和10000张测试集。
这个数据集一共分为了五组训练集和一组测试集,这样子,每组就有10000张随机组成的图片。虽然是随机的,但是在训练集中,每一类的总数还是5000张。
几点需要注意的事项:
- 训练模式的选择
- 数据的下载
- 注意观察输出
具体流程
1 创建各种数据块
cd $CAFFE_ROOT
./data/cifar10/get_cifar10.sh
./examples/cifar10/create_cifar10.sh
2 训练数据
cd $CAFFE_ROOT
./examples/cifar10/train_quick.sh
3 等待最终结果
注意:
- 模式的选择
solver mode: CPU or GPU
solver_mode: CPU