硕士毕业论文第二点扯到了CNN上,准备用caffe框架来做,装好windows下caffe后运行了第一个经典例子LeNet :参考:http://blog.csdn.net/tianrolin/article/details/51434687
运行第二个cifar10例子时,不会写bat文件,运行也各种错误,这里总结一下我遇到的问题和解决办法。亲测有用!
一、数据集准备
到官网中下载cifar10数据集,如果使用的是c++,那么下载.bin文件。数据集介绍已经很详细了。cifar10数据集官网下载
下载并解压,看到共有6个.bin文件,前面五个是训练集,最后一个是测试集。
二、将数据集转成leveldb格式
用vs2013打开项目工程。有一个project是convert_cifar_data。(若没有,则可以在Solution新建一个cuda project,导入tools文件夹中的convert_cifar_data.cpp文件,配置环境即可)。
编译这个project,之后会在../../examples/cifar10/文件夹下找到convert_cifar_data.exe。
新建两个文件夹,分别命名为input_folder,output_folder。input_folder中放入第一阶段准备好的cifar10数据。output_folder为转换数据之后存储位置。
同级目录下新建一个批处理文件,后缀名.bat,输入以下内容并保存。