caffe实现Alexnet网络结构(利用自己的数据)

1.新建测试集和校验集的txt文档 文档内容为:路径+标签

2.修改网络参数,输入数据格式改为图片格式(默认格式为lmdb)

3.执行脚本文件,开始运行

备注:如果使用均值文件,需要输入图片的大小为256*256;均值文件我使用的是imagenet的均值文件,并没有自己生成。


1.新建测试集和校验集的txt文档 文档内容为:路径+标签

1.1在data/目录下,新建mydata文件夹;

1.2在mydata文件夹中新建两个文件夹train var;

1.3在train文件夹中新建n个文件夹(n代表类别数,比如有1000类图像就新建1000个),每个文件夹中存放相同类别      的用于测试的图像库文件

1.4在var中存放用于校验的图像库文件(不区分类别!!!)

1.5生成两个txt文档(train.txt  var.txt),内容为:路径+标签(建议采用全路径;另外txt的生成我是用matlab实现          的,也可以利用caffe提供的指导文 档)

caffe实现Alexnet网络结构(利用自己的数据)_第1张图片

caffe实现Alexnet网络结构(利用自己的数据)_第2张图片


2.修改网络参数,输入数据格式改为图片格式(默认格式为lmdb)

2.1在examples/目录下新建mytask文件夹

2.2将caffe-master/model/bvlc_reference_caffenet中的所有文件拷贝到mytask目录下,并对train_val.prototxt进行修改;修        改如下

caffe实现Alexnet网络结构(利用自己的数据)_第3张图片


3.执行脚本文件,开始运行

sudo  ./exmaples/mytask/train_caffenet.sh

因为我修改了solver.prototxt文件,所以只迭代1000次就出结果啦~

caffe实现Alexnet网络结构(利用自己的数据)_第4张图片

你可能感兴趣的:(开发)