caffe训练CIFAR数据库

CIFAR-10是一个用于普适物体识别的数据集。Cifar-10由60000张32*32的RGB彩色图片构成，50000张训练图片，10000张测试图片，分为10类。cifar下载地址： http://www.cs.toronto.edu/~kriz/cifar.html

数据集分为3个版本，分别是Matlab、python和二进制格式的，这里选择二进制格式的下载。包含五个训练文件，一个测试文件：

1. cifar二进制数据库转换成lmdb文件

新建一个binToLmdb.bat的脚本文件，输入一下内容：

D:\Software\Caffe\caffe-master\Build\x64\Release\convert_cifar_data.exe

D:\TestData\cifar-10-batches-bin D:\TestData\cifar-10-batches-bin lmdb

pause

第一个参数是caffe中已经编译好的cifar数据库转换文件，第二个参数是cifar数据库存放路径，第三个参数是生成的lmdb数据库文件存放路径，第四个参数是转换的目标格式。

双击inToLmdb.bat文件，生成cifar的训练和测试数据：

2.计算训练数据的均值文件

新建一个mean.bat脚本文件，输入以下内容：

D:\Software\Caffe\caffe-master\Build\x64\Release\compute_image_mean.exe

D:\TestData\cifar-10-batches-bin\cifar10_test_lmdb

D:\TestData\cifar-10-batches-bin\cifarMean.binaryproto

pause

双击执行，在目标路径下生成cifar训练数据集的均值文件cifarMean.binaryproto文件。

3. 执行训练

对.\examples\cifar10下的 “cifar10_full_solver.prototxt”文件和“cifar10_full_train_test.prototxt”文件做一些修改，用于下边的训练，修改过程略了。

新建一个train.bat脚本文件，输入以下内容：

D:\Software\Caffe\caffe-master\Build\x64\Release\caffe.exe train

--solver=D:\Software\Caffe\caffe-master\examples\cifar10\cifar10_full_solver.prototxt

pause

双击文件，开始训练，执行结果，训练精度大概是78%：

巴特西

caffe训练CIFAR数据库

最新文章

热门文章