上文 说到关于环境的配置,这次我们聊一聊关于使用vs2015生成后的exe来训练自己的数据。好了,废话不多说,下面开始
一、准备工作
既然要训练自己的数据,那就要准备训练的素材,尽可能多的素材用来训练,然后用部分素材来作为验证。
首先:在caffe的解压目录 data/下新建 文件夹mine(自定义) ,然后在这个文件夹下新建三个文件夹:分别是mean、train、val
train是用来存放训练数据的文件夹,val是用来存放测试数据的文件夹,mean是存放均值文件的文件夹,还有训练的结果。
在train文件夹下放入要训练的图片(若有多种类型,在此文件夹下新建多个子文件夹存放每一种类型的图片),val同理。
其中在文件夹下把图片名称命名为统一格式,val也是。如下:
二、各种文件配置及执行
1.新建在mine目录下新建train.txt问价,用来存放训练图片的名称,要确保在train文件夹下按照这个路径能找到此图片,后面的0 是指种类,若是有多种命名为0,1,2,3...(从0开始) ;新建val.txt 文件,存放测试图片的名称。数据量较大可以写一段程序来完成这项工作。内容栗子如下:
2.在 caffe目录下 \caffe\examples\mnist 复制 lenet_solver.prototxt 和 lenet_train_test.prototxt 文件分别重命名为 solver.prototxt 和 train_val.prototxt
① 对于train_val.prototxt 文件,修改上半部分文件为:
name: "AlexNet"
layer { # 关于训练数据
name: "data"
type: "Data"
top: "data"
top: "label"
include {
p