Caffe是目前产品化最多的库,图像的问题用Coffe很方便,训练只需写prototxt。
一、Caffe使用方法1
这种训练方式不用写代码,这里以图像数据和卷积神经网络模型为例,训练过程如下:
1、Resize图片,转换存储格式(LMDB/LevelDB);
2、定义网络结构(编辑prototxt);
3、定义solver(寻优方法,编辑另一个prototxt);
4、一行命令开始训练(可以基于已有的权重赋值)。
整体过程如下图所示:
第一步:转换格式
使用命令行:convert_imageset,需要输入图片的位置和标签:
如下图所示:
第二步:定义层次结构
以下内容在prototxt中编辑:
数据层:
<