caffe深度学习记录之二:训练自己的数据

上文 说到关于环境的配置,这次我们聊一聊关于使用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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值