CIFAR-10 训练
Cifar-10是由Hinton的两个大弟子Alex Krizhevsky、Ilya Sutskever收集的一个用于普适物体识别的数据集。
1.数据集的准备
本实验使用的数据集是CIFAR-10,一共有60000张32*32的彩色图像,其中50000张是训练集,另外10000张是测试集。数据集共有10个类别,分别如下所示
下载数据集
./data/cifar10/get_cifar10.sh #下载二进制的cifar,并解压,会在/data/cifar10中出现很多batch文件
./examples/cifar10/create_cifar10.sh #运行后将会在examples中出现数据集
2.模型
CIFAR-10的卷积神经网络模型由卷积层,pooling层,ReLU,非线性变换层,局部对比归一化线性分类器组成。该模型定义在caffe/examples/cifar10/cifar10_quick_train_test.prototxt中。
3.训练”quick”模型
我们使用CPU进行训练,所以将文件caffe/examples/cifar10/cifar10_quick_solver.prototxt<

本文介绍了CIFAR-10数据集的准备,包括50000张训练图像和10000张测试图像。通过Caffe构建的卷积神经网络模型进行训练,得到了75%的准确率。同时,文章还讨论了如何使用Python接口进行模型预测,并解决了classify.py文件中均值计算错误和结果显示的问题。
最低0.47元/天 解锁文章
7846

被折叠的 条评论
为什么被折叠?



