caffe入门教程
(一)简介
1.caffe是一个清晰而高效的深度学习框架,纯粹的C++/CUDA架构,支持命令行、Python和MATLAB接口,可以在CPU和GPU直接无缝切换;
2.caffe的主要优势:
(1)CPU与GPU的无缝切换;
(2)模型与优化都是通过配置文件来设置,无需代码;
3.caffe的下载与安装:
(1)下载
(2)安装
(3)caffe的下载与安装以及一些基本的介绍官网已经描述地比较详细,这里不再重复;
4.caffe的使用接口有命令行,python跟matlab,个人觉得训练模型的时候使用命令行已经足够简单了,至于训练好的模型可以使用python与matlab的接口进行调用,本文先描述基于命令行的模型训练,以LeNet模型为例;
(二)LeNet模型
PS:LeNet是手写数字库MNIST上应用比较经典的模型,具有7层网络结构,分别是卷积-下采样-卷积-下采样-全连-全连-分类层,具体网络细节可以参考文章:
Gradient based learning applied to document recognition
1.安装编译完caffe后,其主目录下有:
2.训练模型之前需要先准备好训练数据MNIST,执行以下命令可以下载MNIST数据库:
3.由于caffe支持的数据类型不包括图像类型,所以常规做法需要将图像类型转为lmdb类型: