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提供的指导文 档)




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

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

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


caffe:用自己的图像数据训练模型     ***** **



猜你喜欢

转载自blog.csdn.net/tony2278/article/details/80291097