Caffe for Windows 训练cifar10

我们学习Caffe提供的简单例程,目的是为了让初学者轻松上手,以examples/cifar10/为例,主要用于小图片的分类。

1 cifar10数据集

60000张32*32彩色图片,50000张训练,10000张测试

下载cifar10数据集:http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz

将下载好并解压好的数据存放于/data/cifar10/路径

下载好的原始数据是BINARY(二进制)格式的,需要转换成LMDB或LEVELDB格式才能被Caffe识别(本文以LEVELDB格式为例)。

2 转换格式

我们已经编译好Caffe,可参考前面的博客Caffe环境搭建,只需要在根目录D:\caffe-master下编写一个脚本,完成后双击该脚本即可,我是将Caffe的源码解压在D盘的。

脚本文件如下所示:

.\Build\x64\Release\convert_cifar_data.exe ./data/cifar10 ./examples/cifar10 leveldb
pause

.\Build\x64\Release\convert_cifar_data.exe

./data/cifar10         表示输入数据文件路径

./examples/cifar10  表示输出数据文件路径

leveldb                   表示数据格式,你也可以尝试生成lmdb格式

提示:如何编写bat脚本文件?

答:其实就是新建一个文本文件,然后将后缀名改写成bat就可以。

关于脚本里面的Release模式,有人肯定会问,我用Debug模式不可以吗?

答:也可以。但因为在Debug模式下会出现一系列的问题,我个人建议还是在Release模式下生成Caffe。

3 图像数据均值

我们已经编译好Caffe,只需要在目录D:\caffe-master下编写一个脚本就好,完成后双击脚本。

脚本如下所示:

.\Build\x64\Release\compute_image_mean.exe -backend=leveldb ./examples/cifar10/cifar10_train_leveldb mean.binaryproto
pause

.\Build\x64\Release\compute_image_mean.exe    表示图像数据进行初始化处理,需要compute_image_mean.exe可执行文件

-backend=leveldb  表示数据格式,如果不添加这句话的,默认转化为lmdb

./examples/cifar10/cifar10_train_leveldb 表示刚才生成的训练数据集的路径,而不是val验证数据集的路径

mean.binaryproto 表示输出均值文件名,后缀名为binaryproto。这里默认保存在当前路径下,也就是在D:\caffe-master。

4 配置网络文件

该版本的网络配置文件有很多,我这里使用/examples/cifar10/cifar10_quick_train_test.prototxt。打开之后修改训练数据和验证数据以及均值文件的路径,如下图所示,红色部分为修改过后的:

由于我是CPU模式的,所以还需要修改一下/examples/cifar10/cifar10_quick_solver.prototx文件里面的模式,如下图所示,红色圆圈为修改过的,原先为GPU模式。

5 训练日志

由于使用了cifar10_quick_train_test.prototxt,对应的在模型训练设置文件上我们使用cifar10_quick_solver.prototxt,其实不需要修改内容,当然你也可以自己指定一些参数。

我们已经编译好Caffe,只需要在根目录D:\caffe-master下编写一个脚本,完成后双击该脚本即可。

.\Build\x64\Release\caffe.exe train --solver=examples/cifar10/cifar10_quick_solver.prototxt
pause

要是想进行验证的话,可以修改命令为:

.\Build\x64\Release\caffe.exe test --model=examples/cifar10/cifar10_quick_train_test.prototxt
pause

运行结果,图中红色处为准确率,我们训练测试主要看的也就是它:

时间: 2024-08-03 19:41:00

Caffe for Windows 训练cifar10的相关文章

caffe for windows 训练cifar10 及若干问题

声明:有部分内容转载至笑笑魔导师 1,准备cifar10数据库 cifar10数据库由60000张32*32 彩色图片 共10类 其中50000张训练 10000张测试 下载cifar10数据库:http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz 这是binary格式的,所以我们要把它转换成leveldb格式. 2,转换数据 在../caffe-windows/examples/cifar10文件夹中有一个 convert_cifar_d

caffe在windows编译工程及运行mnist数据集测试

caffe在windows上的配置和编译可以参考如下的博客: http://blog.csdn.net/joshua_1988/article/details/45036993 http://blog.csdn.net/joshua_1988/article/details/45048871 http://blog.csdn.net/dongb5lz/article/details/45171187 http://m.blog.csdn.net/blog/thesby/43535619 按照上面

caffe再见之训练自己的数据

Caffe的数据格式采用leveldb或者lmdb格式 本文采用数据为已标定过的彩色图像,共1000张训练图共10个类别,200张测试图像10个类别,下载地址:http://pan.baidu.com/s/1hsvz4g8. 第一步:数据格式转换 1.编译conver_imageset,在\Caffe-Master\Build\x64\Release下生成convert_imageset.exe. 2.在根目录data文件下新建属于自己的数据集文件夹(主要是为了便于整理,具体位置可以根据自己需要

DeepLearning (五) 基于Keras的CNN 训练cifar-10 数据库

数据库介绍 开发工具 网络框架 训练结果 训练要点 激活函数 Dropout 的作用 训练代码 [原创]Liu_LongPo 转载请注明出处[CSDN]http://blog.csdn.net/llp1992 数据库介绍 Cifar-10是由Hinton的两个大弟子Alex Krizhevsky.Ilya Sutskever收集的一个用于普适物体识别的数据集. Cifar-10由60000张32*32的RGB彩色图片构成,共10个分类.50000张训练,10000张测试(交叉验证).这个数据集最

CNN训练Cifar-10技巧

关于数据集 Cifar-10是由Hinton的两个大弟子Alex Krizhevsky.Ilya Sutskever收集的一个用于普适物体识别的数据集.Cifar是加拿大牵头投资的一个先进科学项目研究所. 说白了,就是看你穷的没钱搞研究,就施舍给你.Hinton.Bengio和他的学生在2004年拿到了Cifar投资的少量资金,建立了神经计算和自适应感知项目. 这个项目结集了不少计算机科学家.生物学家.电气工程师.神经科学家.物理学家.心理学家,加速推动了DL的进程.从这个阵容来看,DL已经和M

基于英特尔® 至强 E5 系列处理器的单节点 Caffe 评分和训练

原文链接 在互联网搜索引擎和医疗成像等诸多领域,深度神经网络 (DNN) 应用的重要性正在不断提升. Pradeep Dubey 在其博文中概述了英特尔? 架构机器学习愿景. 英特尔正在实现 Pradeep Dubey 博文中勾勒的机器学习愿景,并正在着手开发软件解决方案以加速执行机器学习工作负载.这些解决方案将包含在未来版本的英特尔? 数学核心函数库(英特尔? MKL)和英特尔? 数据分析加速库(英特尔? DAAL)中. 本技术预览版展示了配备我们正在开发的软件后,英特尔平台将有望实现的性能.

【caffe】用训练好的imagenet模型分类图像

因为毕设需要,我首先是用ffmpeg抽取某个宠物视频的关键帧,然后用caffe对这个关键帧中的物体进行分类. 1.抽取关键帧的命令: E:\graduation design\FFMPEG\bin>ffmpeg -i .\3.mp4 -vf select='eq(pict_type\,I)',setpts='N/(25*TB)' .\%09d.jpg 2.用python编写脚本,利用在imagenet上训练的模型分类视频帧中的物体. 抽取得到的视频关键帧都存放在文件夹"/home/suns

caffe 在 windows 使用

以前打开windows版本的caffe,是有一个.sln 文件,通过 vs2013 加载后可以直接执行编译过程,这次遇到一些坑,最终还是编译好,并且把 mnist 的例程再次在windows 平台跑通了,应该说,如果不是想用 VS 管理工程写C++ ,用windows真的是蛋疼.'''' 主要步骤如下: 1.下载安装好 Anaconda2.Git.cmake( > 3.4 ). 2.在github中搜索caffe,将分支由 master 改为 windows,下载 仓库到 桌面,可以看到Scri

caffe 利用VGG训练自己的数据

写这个是因为有童鞋在跑VGG的时候遇到各种问题,供参考一下. 网络结构 以VGG16为例,自己跑的细胞数据 solver.prototxt: net: "/media/dl/source/Experiment/cell/test/vgg/vgg16.prototxt" test_iter: 42 test_interval: 1000 base_lr: 0.0001 lr_policy: "step" gamma: 0.1 stepsize: 100000 disp